forked from sawlani/GLAM
-
Notifications
You must be signed in to change notification settings - Fork 0
ZhongLIFR/GLAM
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
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
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%