-
Notifications
You must be signed in to change notification settings - Fork 0
/
arg.py
50 lines (44 loc) · 2.42 KB
/
arg.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--seed', type=int, default = 1)
parser.add_argument('--dataset', type=str, default='WikiCS')
parser.add_argument('--gpu_num', type=str, default='cuda:0')
parser.add_argument('--linear', type=bool, default=False)
parser.add_argument('--embedding_train', type=bool, default = True)
parser.add_argument('--teacher_eval', type=bool, default = False)
parser.add_argument('--node_p', type=float, default=0.7)
parser.add_argument('--edge_p', type=float, default=0.7)
parser.add_argument('--epoch', type=int, default=10000)
parser.add_argument('--lrdec_1', type=float, default=0.8)
parser.add_argument('--lrdec_2', type=int, default=200)
parser.add_argument('--lr', type=float, default=0.01)
parser.add_argument('--dropout', type=float, default=0.6)
parser.add_argument('--test', type=int, default=10000)
parser.add_argument('--tau', type=float, default=0.7)
parser.add_argument('--hidden_num', type=int, default = 64)
parser.add_argument('--head', type=int, default = 4)
parser.add_argument('--out_channels', type=int, default = 512)
parser.add_argument('--write', type=int, default = 0)
parser.add_argument('--ratio', type=float, default=0.3)
parser.add_argument('--batch_size', type=int, default=128)
parser.add_argument('--patience', type=int, default = 200)
parser.add_argument('--embedding_dim', type=int, default = 64)
parser.add_argument('--walk_length', type=int, default = 5)
parser.add_argument('--context_size', type=int, default = 5)
parser.add_argument('--walks_per_node', type=int, default = 5)
parser.add_argument('--node2vec_batchsize', type=int, default = 512)
parser.add_argument('--node2vec_epoch', type=int, default = 100)
parser.add_argument('--node2vec_lr', type=float, default = 0.01)
parser.add_argument('--node2vec_neg_samples', type=int, default = 5)
parser.add_argument('--p', type=float, default = 1.0)
parser.add_argument('--q', type=float, default = 1.0)
parser.add_argument('--l1_e', type=int, default = 4)
parser.add_argument('--l2_e', type=int, default = 1)
parser.add_argument('--l12_e', type=int, default = 1)
parser.add_argument('--l1_f', type=int, default = 1)
parser.add_argument('--l2_f', type=int, default = 4)
parser.add_argument('--l12_f', type=int, default = 1)
parser.add_argument('--l1_b', type=int, default = 2)
parser.add_argument('--l2_b', type=int, default = 2)
parser.add_argument('--l12_b', type=int, default = -1)
args = parser.parse_args()