diff options
Diffstat (limited to 'experiments/run_depth_extras.py')
| -rw-r--r-- | experiments/run_depth_extras.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/experiments/run_depth_extras.py b/experiments/run_depth_extras.py index 66a7d45..e3b860b 100644 --- a/experiments/run_depth_extras.py +++ b/experiments/run_depth_extras.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 """H11: Fill depth sweep at L=14 and L=18 to densify Fig 4(a). -3 methods (BP / DFA-GNN / GRAFT) × 4 datasets × 2 depths × 20 seeds = 480 runs. +3 methods (BP / DFA-GNN / KAFT) × 4 datasets × 2 depths × 20 seeds = 480 runs. """ import torch @@ -8,7 +8,7 @@ import numpy as np import json import os from src.data import load_dataset -from src.trainers import BPTrainer, DFAGNNTrainer, GraphGrAPETrainer +from src.trainers import BPTrainer, DFAGNNTrainer, KAFTTrainer from run_dblp_depth import load_dblp device = 'cuda:0' @@ -24,7 +24,7 @@ dfagnn_extra = dict(diffusion_alpha=0.5, diffusion_iters=10, max_topo_power=3) METHODS = { 'BP': (BPTrainer, {}), 'DFA-GNN': (DFAGNNTrainer, dfagnn_extra), - 'GRAFT': (GraphGrAPETrainer, grape_extra), + 'KAFT': (KAFTTrainer, grape_extra), } |