From ba6ead6d7a41b7ed78bb228181b7262d0c75d2eb Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Mon, 4 May 2026 23:10:10 -0500 Subject: =?UTF-8?q?Global=20rename=20GRAFT=20=E2=86=92=20KAFT=20(incl.=20i?= =?UTF-8?q?nternal=20class=20+=20filenames)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 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. --- experiments/run_depth_extras.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'experiments/run_depth_extras.py') 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), } -- cgit v1.2.3