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 /figures/gen_fig4_combined.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 'figures/gen_fig4_combined.py')
| -rw-r--r-- | figures/gen_fig4_combined.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/figures/gen_fig4_combined.py b/figures/gen_fig4_combined.py index 5b8d464..1cdee10 100644 --- a/figures/gen_fig4_combined.py +++ b/figures/gen_fig4_combined.py @@ -3,7 +3,7 @@ Each panel: 9 curves = 3 datasets × 3 methods. color = dataset (Cora / CiteSeer / PubMed) - linestyle = method (BP dashed, DFA-GNN dotted, GRAFT solid) + linestyle = method (BP dashed, DFA-GNN dotted, KAFT solid) Matches DFA-GNN Figure 4 layout. """ @@ -15,18 +15,18 @@ from matplotlib.colors import to_rgba from matplotlib.lines import Line2D DATASETS = ['Cora', 'CiteSeer', 'PubMed'] -METHODS = ['BP', 'DFA-GNN', 'GRAFT'] # data-lookup keys (unchanged) -DISPLAY_NAME = {'BP': 'BP', 'DFA-GNN': 'DFA-GNN', 'GRAFT': 'KAFT'} +METHODS = ['BP', 'DFA-GNN', 'KAFT'] # data-lookup keys (unchanged) +DISPLAY_NAME = {'BP': 'BP', 'DFA-GNN': 'DFA-GNN', 'KAFT': 'KAFT'} DEPTHS = [4, 6, 8, 10, 12, 14, 16, 18, 20] RATES = [0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8] ATTACKS = ['add', 'remove', 'flip'] -# Method colors — consistent with other GRAFT figures +# Method colors — consistent with other KAFT figures METHOD_COLORS = { 'BP': '#888888', # gray 'DFA-GNN': '#3B7AC2', # complementary blue - 'GRAFT': '#C23B3B', # brick red (ours) + 'KAFT': '#C23B3B', # brick red (ours) } # Dataset linestyles DS_STYLE = { @@ -48,7 +48,7 @@ DEPTH_SOURCES = [ 'results/combo_20seeds/per_seed_data.json', 'results/hero_extras_20seeds/per_seed_data.json', 'results/shallow_depth_20seeds/per_seed_data.json', - 'results/bp_graft_depth_20seeds/per_seed_data.json', + 'results/bp_kaft_depth_20seeds/per_seed_data.json', 'results/dfagnn_depth_20seeds/per_seed_data.json', 'results/dfagnn_resgcn_20seeds/per_seed_data.json', 'results/depth_extras_20seeds/per_seed_data.json', # L=14, 18 |