forked from sawlani/GLAM
-
Notifications
You must be signed in to change notification settings - Fork 0
/
README.txt
26 lines (15 loc) · 1.24 KB
/
README.txt
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
Prerequisites: torch, torch-geometric
Example commands to run a single model configuration,
python main.py --data=PROTEINS --lr=0.1
python main.py --data=DHFR --use_node_attr --ignore_node_labels
python main.py --data=COLLAB --aggregation=Mean --lr=0.001
To run a number of configurations, to use for model selection later, set the required configurations in config.txt and run:
python main.py --use_config
Once the pickle file is generated for outputs of all configurations, use model selection to select models. If you use "--aggergation=both", make sure that both files GIN_MMD_<dataset>_<seed>.pkl and GIN_Mean_<dataset>_<seed>.pkl have been generated in the output folder. Here are two examples:
python model_selection.py --data=PROTEINS --seed=1213 --aggregation=MMD
python model_selection.py --data=AIDS --aggregation=both
Run both files with --help to see all parameters that can be input.
Here is a chunk of code which should run all experiments needed to recreate results in GLAM:
python main.py --data=PROTEINS --use_config --config_file=config/MMD_config.txt
python main.py --data=PROTEINS --aggregation=Mean --use_config --config_file=config/Mean_config.txt
python model_selection.py --data=PROTEINS