|
|
|
@ -24,16 +24,13 @@ def parser_args():
|
|
|
|
|
parser.add_argument("-d", "--dataset", type=str, default="Beauty", help="choose which dataset")
|
|
|
|
|
parser.add_argument("-dpath", "--datapath", type=str, default="./scripts/data_mr", help="minddata path")
|
|
|
|
|
parser.add_argument("-de", "--device", type=str, default='0', help="device id")
|
|
|
|
|
parser.add_argument('--seed', type=int, default=0, help="random seed")
|
|
|
|
|
parser.add_argument('--Ks', type=list, default=[5, 10, 20, 100], help="top K")
|
|
|
|
|
parser.add_argument('--test_ratio', type=float, default=0.2, help="test ratio")
|
|
|
|
|
parser.add_argument('-w', '--workers', type=int, default=8, help="number of process")
|
|
|
|
|
parser.add_argument('-w', '--workers', type=int, default=8, help="number of process to generate data")
|
|
|
|
|
parser.add_argument("-ckpt", "--ckptpath", type=str, default="./ckpts", help="checkpoint path")
|
|
|
|
|
|
|
|
|
|
parser.add_argument("-eps", "--epsilon", type=float, default=1e-8, help="optimizer parameter")
|
|
|
|
|
parser.add_argument("-lr", "--learning_rate", type=float, default=1e-3, help="learning rate")
|
|
|
|
|
parser.add_argument("-l2", "--l2", type=float, default=0.03, help="l2 coefficient")
|
|
|
|
|
parser.add_argument("-wd", "--weight_decay", type=float, default=0.01, help="weight decay")
|
|
|
|
|
parser.add_argument("-act", "--activation", type=str, default='tanh', choices=['relu', 'tanh'],
|
|
|
|
|
help="activation function")
|
|
|
|
|
parser.add_argument("-ndrop", "--neighbor_dropout", type=list, default=[0.0, 0.2, 0.3],
|
|
|
|
@ -52,7 +49,4 @@ def parser_args():
|
|
|
|
|
parser.add_argument("-emb", "--embedded_dimension", type=int, default=64, help="output embedding dim")
|
|
|
|
|
parser.add_argument('--dist_reg', type=float, default=0.003, help="distance loss coefficient")
|
|
|
|
|
|
|
|
|
|
parser.add_argument('-ng', '--num_graphs', type=int, default=5, help="num of sample graph")
|
|
|
|
|
parser.add_argument('-geps', '--graph_epsilon', type=float, default=0.01, help="node copy parameter")
|
|
|
|
|
|
|
|
|
|
return parser.parse_args()
|
|
|
|
|