| { | |
| "paths": { | |
| "dataset": "weight/datasets/PRO_Test60.csv", | |
| "feature_dir": "weight/feature", | |
| "weight_dir": "weight", | |
| "output_dir": "output/prediction" | |
| }, | |
| "runtime": { | |
| "device": "auto", | |
| "num_workers": 8, | |
| "seed": 2024 | |
| }, | |
| "model": { | |
| "node_features": 1038, | |
| "edge_features": 16, | |
| "hidden_dim": 128, | |
| "num_encoder_layers": 4, | |
| "k_neighbors": 30, | |
| "augment_eps": 0.1, | |
| "dropout": 0.3, | |
| "batch_size": 4, | |
| "folds": 5, | |
| "obj_max": 1, | |
| "epochs": 30, | |
| "patience": 8, | |
| "num_samples": 1675 | |
| } | |
| } | |