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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
"""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()
|
