conf/config.json

{
  "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
  }
}