best_ensemble.json

{
  "name": "config",
  "is_kaggle": false,
  "log_to_wandb": true,
  "pred_with_cpu": false,
  "train_series_path": "data/raw/train_series.parquet",
  "train_events_path": "data/raw/train_events.csv",
  "test_series_path": "data/raw/test_series.parquet",
  "fe_loc_in": "data/processed",
  "processed_loc_out": "data/processed",
  "processed_loc_in": "data/raw",
  "model_store_loc": "tm",
  "model_config_loc": "model_configs",
  "ensemble": {
    "models": [
        "spectrogram-cnn-gru.json",
        "best/spectre-simple.json",
        "best/STFGO.json",
        "best/STFGO-simple.json",
        "best/STFGO-sweep.json",
        "gruo_best/minion1.json",
        "gruo_best/minion2.json",
        "gruo_best/minion4.json",
        "best/horsemen1.json"
    ],
    "weights": [
        1,
        1,
        1,
        1,
        1,
        1,
        1,
        1,
        1
    ],
    "comb_method": "confidence_average",
    "pred_only": false
  },
  "cv": {
    "splitter": "group_k_fold",
    "scoring": [
      "score_full",
      "score_clean"
    ],
    "splitter_params": {
      "n_splits": 5
    }
  },
  "train_for_submission": false,
  "scoring": true,
  "visualize_preds": {
    "n": 0,
    "browser_plot": false,
    "save": true
  }
}