config.json

{
  "model_type": "kumo_relational",
  "checkpoint_files": {
    "classification": "classifier.pt",
    "regression": "regressor.pt"
  }
}