-
Notifications
You must be signed in to change notification settings - Fork 1
/
parse.py
42 lines (38 loc) · 2.16 KB
/
parse.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
def parser_add_main_args(parser):
# setup and protocol
parser.add_argument('--dataset', type=str, default='cora')
parser.add_argument('--data_dir', type=str,
default='/mnt/nas/home/niefan/ODgraph-energy/data/')
parser.add_argument('--device', type=int, default=0,
help='which gpu to use if any (default: 0)')
parser.add_argument('--cpu', action='store_true')
parser.add_argument('--seed', type=int, default=123)
parser.add_argument('--runs', type=int, default=5,
help='number of distinct runs')
parser.add_argument('--epochs', type=int, default=500)
# model network
parser.add_argument('--hidden_channels', type=int, default=64)
parser.add_argument('--num_layers', type=int, default=2,
help='number of layers for deep methods')
# CaNet
parser.add_argument('--backbone_type', type=str, default='gcn', choices=['gcn', 'gat'])
parser.add_argument('--K', type=int, default=3,
help='num of domains, each for one graph convolution filter')
parser.add_argument('--tau', type=float, default=1,
help='temperature for Gumbel Softmax')
parser.add_argument('--env_type', type=str, default='node', choices=['node', 'graph'])
parser.add_argument('--lamda', type=float, default=1.0,
help='weight for regularlization')
parser.add_argument('--variant', action='store_true',help='set to use variant')
# training
parser.add_argument('--weight_decay', type=float, default=5e-4)
parser.add_argument('--dropout', type=float, default=0.0)
parser.add_argument('--lr', type=float, default=0.01)
parser.add_argument('--use_bn', action='store_true', help='use batch norm')
# display and utility
parser.add_argument('--display_step', type=int,
default=1, help='how often to print')
parser.add_argument('--store_result', action='store_true',
help='whether to store results')
parser.add_argument('--combine_result', action='store_true',
help='whether to combine all the ood environments')