-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathConfig-Examples.sh
More file actions
7 lines (4 loc) · 981 Bytes
/
Copy pathConfig-Examples.sh
File metadata and controls
7 lines (4 loc) · 981 Bytes
1
2
3
4
5
6
7
# run the following commands to train and evaluate models with example configurations on 3 demo datasets
python main.py --dataset Cora --epochs 200 --model argnn --optimizer adam --lr 0.005 --num_runs 10 --argnn_hidden_dim 128 --argnn_num_layers 3 --argnn_metric_hidden_dim 128 --argnn_metric_reg 0.01 --argnn_smoothness_reg 0.001 --dropout 0.3 --task node_classification --logs ./logs
python main.py --dataset Citeseer --epochs 200 --model argnn --optimizer adam --lr 0.005 --num_runs 10 --argnn_hidden_dim 128 --argnn_num_layers 3 --argnn_metric_hidden_dim 128 --argnn_metric_reg 0.01 --argnn_smoothness_reg 0.001 --dropout 0.2 --task node_classification --logs ./logs
python main.py --dataset PubMed --epochs 200 --model argnn --optimizer adam --lr 0.005 --num_runs 10 --argnn_hidden_dim 128 --argnn_num_layers 3 --argnn_metric_hidden_dim 128 --argnn_metric_reg 0.01 --argnn_smoothness_reg 0.001 --dropout 0.2 --task node_classification --logs ./logs