config.json

{
  "input_dim": 77,
  "hidden_dims_a": [
    32,
    16
  ],
  "hidden_dims_b": [
    128,
    64
  ],
  "dropout_a": 0.2,
  "dropout_b": 0.0,
  "threshold_a": 0.60,
  "threshold_b": 0.47421052631578947,
  "loss_a": "PID",
  "loss_b": "PID",
  "hp_a": {
    "hidden_dims": [
      32,
      16
    ],
    "lr": 0.001,
    "beta_kl": 0.02,
    "dropout": 0.2,
    "weight_decay": 0.0001,
    "use_pid_loss": true
  },
  "hp_b": {
    "hidden_dims": [
      128,
      64
    ],
    "lr": 0.001,
    "beta_kl": 0.02,
    "dropout": 0.0,
    "weight_decay": 0.0001,
    "use_pid_loss": true
  },
  "local_model": "Qwen/Qwen3-Coder-Next",
  "sonnet_model": "claude-sonnet-4-6",
  "opus_model": "claude-opus-4-6",
  "n_train": 1150,
  "n_val": 247,
  "n_test": 247,
  "test_results": {
    "local_rate": 0.0,
    "sonnet_rate": 0.2834008097165992,
    "opus_rate": 0.7165991902834008,
    "utility": 0.6205263137817383,
    "oracle_utility": 0.7178542613983154,
    "regret": 0.09732794016599655
  }
}