From c54ddb88b532be28ca3096e21de405d90163ecfa Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Mon, 29 Jun 2026 12:04:47 -0500 Subject: Package full RRoG GNN project --- diag/train_cycle.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'diag/train_cycle.py') diff --git a/diag/train_cycle.py b/diag/train_cycle.py index d2342f3..598e349 100644 --- a/diag/train_cycle.py +++ b/diag/train_cycle.py @@ -24,9 +24,14 @@ from torch_geometric.loader import DataLoader from torch_geometric.utils import to_networkx from torch_geometric.nn import GINConv, GCNConv, global_add_pool -ROOT = '/home/yurenh2/rrog/data/zinc' -CACHE = '/home/yurenh2/rrog/data/cycle_cache' -OUT = '/home/yurenh2/rrog/runs' +PROJECT_ROOT = os.environ.get( + 'RROG_ROOT', + os.path.abspath(os.path.join(os.path.dirname(__file__), '..')), +) +DATA_ROOT = os.environ.get('RROG_DATA_DIR', os.path.join(PROJECT_ROOT, 'data')) +OUT = os.environ.get('RROG_RUNS_DIR', os.path.join(PROJECT_ROOT, 'runs')) +ROOT = os.path.join(DATA_ROOT, 'zinc') +CACHE = os.path.join(DATA_ROOT, 'cycle_cache') RWSE_K = 16 -- cgit v1.2.3