diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-05-04 23:10:10 -0500 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-05-04 23:10:10 -0500 |
| commit | ba6ead6d7a41b7ed78bb228181b7262d0c75d2eb (patch) | |
| tree | 726171fb4b0c536d9287a15daf52929ec65fa3d0 /experiments/run_hero_extras.py | |
| parent | 37ba0f83e3652a215680fd8515af9c14fc02e21c (diff) | |
Global rename GRAFT → KAFT (incl. internal class + filenames)
- src/trainers.py: GraphGrAPETrainer → KAFTTrainer; module docstring + comments.
VanillaGrAPETrainer kept as-is (it is a separate control method, not KAFT).
- experiments/: all 19 runners pick up the new class name; result keys
('Cora_GRAFT' etc) become 'Cora_KAFT'; OUT_DIRs renamed (e.g.
bp_graft_depth_20seeds → bp_kaft_depth_20seeds).
- figures/: data-lookup keys + display labels both 'KAFT'; output filename
graft_depth_sweep.{pdf,png} → kaft_depth_sweep.{pdf,png}.
- File rename: experiments/run_bp_graft_depth.py → run_bp_kaft_depth.py;
figures/graft_depth_sweep.pdf → kaft_depth_sweep.pdf.
- README aligned.
Imports verified: from src.trainers import KAFTTrainer succeeds.
Diffstat (limited to 'experiments/run_hero_extras.py')
| -rw-r--r-- | experiments/run_hero_extras.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/experiments/run_hero_extras.py b/experiments/run_hero_extras.py index c6fed4d..ced76b8 100644 --- a/experiments/run_hero_extras.py +++ b/experiments/run_hero_extras.py @@ -1,8 +1,8 @@ #!/usr/bin/env python3 """E0a+E0c+E0e: hero-table coverage expansion. Adds 3 datasets (PubMed stack, -Coauthor-Physics, Coauthor-CS) × 5 methods (BP, DFA, DFA-GNN, GRAFT, -GRAFT+ResGCN) × 20 seeds, all GCN L=6. Goal: 6-row hero table of homophilous -citation/coauthor graphs where GRAFT or GRAFT+ResGCN is best per row.""" +Coauthor-Physics, Coauthor-CS) × 5 methods (BP, DFA, DFA-GNN, KAFT, +KAFT+ResGCN) × 20 seeds, all GCN L=6. Goal: 6-row hero table of homophilous +citation/coauthor graphs where KAFT or KAFT+ResGCN is best per row.""" import torch import numpy as np @@ -10,7 +10,7 @@ import json import os from scipy import stats as scipy_stats from src.data import load_dataset, spmm -from src.trainers import BPTrainer, DFATrainer, DFAGNNTrainer, GraphGrAPETrainer +from src.trainers import BPTrainer, DFATrainer, DFAGNNTrainer, KAFTTrainer from run_deep_baselines import ResGCNTrainer from run_combo_20seeds import GRAFTResGCN from run_large_graph_scout import load_and_check @@ -54,8 +54,8 @@ METHODS = { 'BP': (BPTrainer, {}), 'DFA': (DFATrainer, dfagnn_extra), 'DFA-GNN': (DFAGNNTrainer, dfagnn_extra), - 'GRAFT': (GraphGrAPETrainer, grape_extra), - 'GRAFT+ResGCN': (GRAFTResGCN, grape_extra), + 'KAFT': (KAFTTrainer, grape_extra), + 'KAFT+ResGCN': (GRAFTResGCN, grape_extra), } |
