| { | |
| "seed": 42, | |
| "data": { | |
| "raw_dir": "data/raw", | |
| "processed_dir": "data/processed", | |
| "shard_size": 2000, | |
| "preprocess_workers": 8, | |
| "train_ratio": 0.8, | |
| "val_ratio": 0.1, | |
| "test_ratio": 0.1 | |
| }, | |
| "features": { | |
| "use_degree": true, | |
| "use_log_degree": true, | |
| "use_normalised_degree": true, | |
| "use_degree_bucket": true, | |
| "degree_bucket_bins": 16, | |
| "use_clustering": true, | |
| "use_kcore": true, | |
| "use_pagerank": true, | |
| "use_laplacian_pe": true, | |
| "laplacian_pe_dim": 8, | |
| "use_rwse": true, | |
| "rwse_steps": 8, | |
| "pagerank_alpha": 0.85, | |
| "pagerank_max_iter": 100 | |
| }, | |
| "model": { | |
| "architecture": "gin", | |
| "hidden_dim": 128, | |
| "num_layers": 4, | |
| "dropout": 0.2, | |
| "num_heads": 4, | |
| "use_virtual_node": true, | |
| "pooling": "attention", | |
| "gin_eps": 0.0, | |
| "pna_aggregators": [ | |
| "mean", | |
| "max", | |
| "min", | |
| "std" | |
| ], | |
| "pna_scalers": [ | |
| "identity", | |
| "amplification", | |
| "attenuation" | |
| ] | |
| }, | |
| "training": { | |
| "batch_size": 4096, | |
| "num_epochs": 40, | |
| "learning_rate": 0.003, | |
| "weight_decay": 0.0001, | |
| "grad_clip": 1.0, | |
| "early_stopping_patience": 8, | |
| "scheduler": "cosine", | |
| "use_amp": false, | |
| "num_workers": 0 | |
| }, | |
| "output": { | |
| "runs_dir": "runs", | |
| "checkpoints_dir": "checkpoints" | |
| } | |
| } |