diff --git a/trainer/utils/config.py b/trainer/utils/config.py new file mode 100644 index 0000000000000000000000000000000000000000..bb9c1ef28d385b80c7673027b4f69b31c36b1175 --- /dev/null +++ b/trainer/utils/config.py @@ -0,0 +1,20 @@ +import argparse + + +def build_args(): + parser = argparse.ArgumentParser(description="MAGIC") + parser.add_argument("--dataset", type=str, default="wget") + parser.add_argument("--device", type=int, default=-1) + parser.add_argument("--lr", type=float, default=0.001, + help="learning rate") + parser.add_argument("--weight_decay", type=float, default=5e-4, + help="weight decay") + parser.add_argument("--negative_slope", type=float, default=0.2, + help="the negative slope of leaky relu for GAT") + parser.add_argument("--mask_rate", type=float, default=0.5) + parser.add_argument("--alpha_l", type=float, default=3, help="`pow`inddex for `sce` loss") + parser.add_argument("--optimizer", type=str, default="adam") + parser.add_argument("--loss_fn", type=str, default='sce') + parser.add_argument("--pooling", type=str, default="mean") + args = parser.parse_args() + return args