"""Argument parsing."""

import argparse

def parameter_parser():
    """
    A method to parse up command line parameters.
    The default hyperparameters give a high performance model without grid search.
    """
    parser = argparse.ArgumentParser(description = "Run SEAL-CI/SEAL-AI.")

    parser.add_argument("--graphs",
                        nargs="?",
                        default="D:\学习\mesa\毕设\研究点三\OpenWGL-main\opgl\dataset\\test1/",
	                help="Folder with training graph jsons.")

    parser.add_argument("--hierarchical-graph",
                        nargs="?",
                        default="D:\学习\mesa\毕设\研究点三\OpenWGL-main\opgl\dataset\\test1\edges_DoS.csv",
	                help="Hierarchical edge list.")

    parser.add_argument("--labeled-count",
                        type=int,
                        default=1000,
	                help="Number of labeled data points. Default is 100.")

    parser.add_argument("--budget",
                        type=int,
                        default=20,
	                help="Number of data points added in learning phase. Default is 20.")

    parser.add_argument("--first-gcn-dimensions",
                        type=int,
                        default=16,
	                help="Filters (neurons) in 1st convolution. Default is 32.")

    parser.add_argument("--second-gcn-dimensions",
                        type=int,
                        default=8,
	                help="Filters (neurons) in 2nd convolution. Default is 16.")

    parser.add_argument("--first-dense-neurons",
                        type=int,
                        default=16,
	                help="Neurons in SAGE aggregator layer. Default is 16.")

    parser.add_argument("--second-dense-neurons",
                        type=int,
                        default=4,
	                help="SAGE attention neurons. Default is 8.")

    parser.add_argument("--macro-gcn-dimensions",
                        type=int,
                        default=16,
	                help="Filters (neurons) in 1st macro convolution. Default is 16.")

    parser.add_argument("--epochs",
                        type=int,
                        default=15,
	                help="Number of epochs. Default is 10.")

    parser.add_argument("--learning-rate",
                        type=float,
                        default=0.01,
	                help="Learning rate. Default is 0.01.")

    parser.add_argument("--weight-decay",
                        type=float,
                        default=5*10**-5,
	                help="Adam weight decay. Default is 5*10^-5.")

    parser.add_argument("--gamma",
                        type=float,
                        default=10**-5,
	                help="Attention regularization coefficient. Default is 10^-5.")
    parser.add_argument("--feature_which",
                        type=list,
                        default=[0, 1, 2, 3, 4, 5],
                        help="Attention regularization coefficient. Default is 10^-5.")

    return parser.parse_args()