diff options
Diffstat (limited to 'experiments/run_cora_perturb.py')
| -rw-r--r-- | experiments/run_cora_perturb.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/experiments/run_cora_perturb.py b/experiments/run_cora_perturb.py index 1dabc95..1124b97 100644 --- a/experiments/run_cora_perturb.py +++ b/experiments/run_cora_perturb.py @@ -13,7 +13,7 @@ import numpy as np import json import os from src.data import load_dataset, build_normalized_adj, build_row_normalized_adj -from src.trainers import BPTrainer, DFATrainer, GraphGrAPETrainer +from src.trainers import BPTrainer, DFATrainer, KAFTTrainer device = 'cuda:0' SEEDS = [0, 1, 2, 3, 4] @@ -111,7 +111,7 @@ def main(): common = dict(data=data, hidden_dim=64, lr=0.01, weight_decay=5e-4, num_layers=L, residual_alpha=0.0, backbone='gcn') bp_accs.append(train_one(BPTrainer, common, {}, seed)) - gr_accs.append(train_one(GraphGrAPETrainer, common, grape_extra, seed)) + gr_accs.append(train_one(KAFTTrainer, common, grape_extra, seed)) bp, gr = np.mean(bp_accs)*100, np.mean(gr_accs)*100 delta = gr - bp |