cases/model_config.json

{
  "model_type": "Mamba2Regressor",
  "predictor": "cases",
  "input_size": 16,
  "d_model": 256,
  "n_layers": 4,
  "d_state": 128,
  "d_conv": 4,
  "expand": 2,
  "head_dim": 64,
  "chunk_size": 256,
  "output_dim": 1,
  "dropout": 0.3,
  "seed": 42
}