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 /README.md | |
| 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 'README.md')
| -rw-r--r-- | README.md | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -1,7 +1,7 @@ # KAFT: Topology-Factorized Jacobian-Aligned Feedback for Deep GNNs -> Internal class names (`GraphGrAPETrainer`, file paths starting with -> `graft_*` / dataset keys `'GRAFT'`) are the original code identifier. +> Internal class names (`KAFTTrainer`, file paths starting with +> `kaft_*` / dataset keys `'KAFT'`) are the original code identifier. > The method is referred to as **KAFT** in the paper. Code release accompanying the NeurIPS 2026 submission. @@ -24,7 +24,7 @@ computed in O(1) parallel depth on GPUs. ``` src/ core method - trainers.py BPTrainer, GraphGrAPETrainer (= KAFT), DFA/DFA-GNN, alignment + trainers.py BPTrainer, KAFTTrainer (= KAFT), DFA/DFA-GNN, alignment data.py PyG dataset loaders, normalized  / row-Â, sparse-mm helpers experiments/ one runner per reported result block (see `## Reproducing`) figures/ figure generators + the four rendered PDFs in the paper @@ -51,7 +51,7 @@ CUDA_VISIBLE_DEVICES=0 python -u experiments/run_ablation_20seeds.py # Fig 2: Planetoid depth sweep (11 / 13 points) CUDA_VISIBLE_DEVICES=0 python -u experiments/run_shallow_depth.py -CUDA_VISIBLE_DEVICES=0 python -u experiments/run_bp_graft_depth.py +CUDA_VISIBLE_DEVICES=0 python -u experiments/run_bp_kaft_depth.py CUDA_VISIBLE_DEVICES=0 python -u experiments/run_dfagnn_depth.py CUDA_VISIBLE_DEVICES=0 python -u experiments/run_depth_extras.py CUDA_VISIBLE_DEVICES=0 python -u experiments/run_dblp_depth_scaling.py |