From 174083f4e340afc192d57aacc6422c99530e9e59 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Mon, 4 May 2026 23:12:29 -0500 Subject: =?UTF-8?q?Drop=20VanillaGrAPE=20=E2=80=94=20control=20method=20no?= =?UTF-8?q?t=20exposed=20in=20release?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - src/trainers.py: removed class VanillaGrAPETrainer (~30 lines) and cleaned the module-level methods-compared docstring. - experiments/run_ablation_20seeds.py: dropped VanillaGrAPE row from the 4-method ablation grid; sweep is now BP → DFA → DFA-GNN → KAFT. Smoke test: BPTrainer / DFATrainer / DFAGNNTrainer / KAFTTrainer all train cleanly at GCN L=4, Cora, 50 epochs (test acc 77.3 / 76.6 / 78.4 / 79.0). --- experiments/run_ablation_20seeds.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) (limited to 'experiments/run_ablation_20seeds.py') diff --git a/experiments/run_ablation_20seeds.py b/experiments/run_ablation_20seeds.py index d2bd434..e90917d 100644 --- a/experiments/run_ablation_20seeds.py +++ b/experiments/run_ablation_20seeds.py @@ -1,5 +1,5 @@ #!/usr/bin/env python3 -"""Ablation study with 20 seeds: BP → DFA → DFA-GNN → VanillaGrAPE → KAFT.""" +"""Ablation study with 20 seeds: BP → DFA → DFA-GNN → KAFT.""" import torch import numpy as np @@ -7,7 +7,7 @@ import json import os from scipy import stats as scipy_stats from src.data import load_dataset -from src.trainers import BPTrainer, DFATrainer, DFAGNNTrainer, VanillaGrAPETrainer, KAFTTrainer +from src.trainers import BPTrainer, DFATrainer, DFAGNNTrainer, KAFTTrainer device = 'cuda:0' SEEDS = list(range(20)) @@ -18,10 +18,6 @@ METHODS = { 'BP': (BPTrainer, {}), 'DFA': (DFATrainer, {}), 'DFA-GNN': (DFAGNNTrainer, {'topo_mode': 'fixed_A'}), - 'VanillaGrAPE': (VanillaGrAPETrainer, { - 'diffusion_alpha': 0.5, 'diffusion_iters': 10, - 'lr_feedback': 0.5, 'num_probes': 64, 'topo_mode': 'fixed_A' - }), 'KAFT': (KAFTTrainer, { 'diffusion_alpha': 0.5, 'diffusion_iters': 10, 'lr_feedback': 0.5, 'num_probes': 64, 'topo_mode': 'fixed_A' -- cgit v1.2.3