{
  "schema_version": 1,
  "onnx": "/Users/afengler/hssm_mlflow_mirror/promote_candidates/angle_extended_net4L_promote_staging/angle_extended_lan_e0a375cea73d11f1abc5cfbfc94f3670_model.onnx",
  "model": "angle_extended",
  "network_type": "lan",
  "passed": true,
  "gates": [
    {
      "gate": "structure",
      "passed": true,
      "input_shape": [
        1,
        7
      ],
      "output_shape": [
        1,
        1
      ],
      "input_width": 7,
      "ops": [
        "Gemm",
        "Tanh"
      ]
    },
    {
      "gate": "parity",
      "passed": true,
      "skipped": true,
      "reason": "no *_train_state.jax + config pair"
    },
    {
      "gate": "hssm_load",
      "passed": true,
      "initial_logp": -606.0779265163969,
      "n_trials": 100
    },
    {
      "gate": "density",
      "passed": true,
      "worst_total_mass": 0.9965389547869563,
      "worst_hellinger": 0.05427714430049023,
      "worst_ratio": 0.9193018775779048,
      "mass_range": [
        0.9,
        1.1
      ],
      "hellinger_ratio_max": 3.0,
      "draws": [
        {
          "theta": [
            1.314832198285961,
            1.15273930172996,
            0.20622305531916463,
            0.22733099473722373,
            0.9508626679043052
          ],
          "total_mass": 0.9991344809532166,
          "hellinger": 0.047180253256785196,
          "sampling_floor": 0.06240083701491786,
          "ratio": 0.7560836603122174
        },
        {
          "theta": [
            3.962453541866128,
            1.8803332756571087,
            0.6468777990297591,
            1.0702650863515042,
            1.0872811146423005
          ],
          "total_mass": 0.9988806816265878,
          "hellinger": 0.022226582646181343,
          "sampling_floor": 0.028579530211093947,
          "ratio": 0.7777098672375472
        },
        {
          "theta": [
            3.032194119566708,
            0.57591516036752,
            0.7287387370160444,
            0.2546100520284986,
            0.8572141000015374
          ],
          "total_mass": 1.001399795524776,
          "hellinger": 0.05427714430049023,
          "sampling_floor": 0.059041698515285145,
          "ratio": 0.9193018775779048
        },
        {
          "theta": [
            -3.1137060422154335,
            2.434466472275755,
            0.5265351809594188,
            0.6801992553473857,
            0.5134096877413775
          ],
          "total_mass": 0.9979720581827678,
          "hellinger": 0.029381484625764464,
          "sampling_floor": 0.0380336576895097,
          "ratio": 0.7725127271645071
        },
        {
          "theta": [
            -4.528131157003555,
            0.8384518772390581,
            0.6091996254039235,
            1.2358854669095407,
            0.7292313248590044
          ],
          "total_mass": 0.9965389547869563,
          "hellinger": 0.029625460449983432,
          "sampling_floor": 0.03675445353050921,
          "ratio": 0.8060372990008372
        }
      ]
    }
  ]
}