te_ped_89/model_config.json

{
  "signal_dim": 32,
  "history_features": 458,
  "history_length": 50,
  "aux_input_dim": 3,
  "mse_channels": 512,
  "mse_blocks": 5,
  "mse_embed_dim": 512,
  "mse_kernel_size": 5,
  "fpe_channels": 512,
  "fpe_blocks": 14,
  "fpe_kernel_size": 7,
  "fpe_use_dilated": true,
  "aux_embed_dim": 64,
  "dropout": 0.26803639559450243,
  "channel_mult": 0.75,
  "use_gradient_checkpointing": false,
  "use_mse": true,
  "forecast_horizon": 0,
  "task_type": "regression",
  "target_name": "te_ped",
  "bundle_name": "te_ped_89",
  "dataset_version": "v2_458",
  "input_contract": {
    "fpe_signal_names": [
      "pohm",
      "pinj",
      "tinj",
      "ech_total",
      "f1a",
      "f2a",
      "f3a",
      "f4a",
      "f5a",
      "f6a",
      "f7a",
      "f8a",
      "f9a",
      "f1b",
      "f2b",
      "f3b",
      "f4b",
      "f5b",
      "f6b",
      "f7b",
      "f8b",
      "f9b",
      "ecoila",
      "ecoilb",
      "gasa_cal",
      "gasb_cal",
      "gasc_cal",
      "gasd_cal",
      "gase_cal",
      "ip",
      "ipspr15v",
      "bt"
    ],
    "fpe_signal_dim": 32,
    "mse_history_width": 458,
    "mse_history_length": 50,
    "aux_feature_names": [
      "bzn_seconds",
      "disrupt_seconds",
      "disrupt_coverage"
    ]
  },
  "target_mean": 0.5160982824407585,
  "target_std": 0.4096386938561064,
  "clip_min": -0.5,
  "clip_max": 30.0,
  "target_units": "keV"
}