train_log.json

[
  {
    "epoch": 1,
    "step": 85,
    "loss": 0.9641271437353948,
    "sigma": 0.4,
    "epoch_seconds": 228.8010711669922,
    "train_rows": 5400,
    "raw_ece": 0.10835536271333696,
    "raw_brier": 0.07556118707742442,
    "raw_nll": 0.8722028755644957,
    "raw_accuracy": 0.7066666666666667,
    "raw_soft_accuracy": 0.4981412665049235,
    "raw_mean_confidence": 0.5995895755290985,
    "raw_mean_target_max": 0.6725333379705747,
    "raw_score_mae": 0.29329004988539964,
    "raw_n": 600
  },
  {
    "epoch": 2,
    "step": 170,
    "loss": 0.8357182100488275,
    "sigma": 0.30000000000000004,
    "epoch_seconds": 224.5660684108734,
    "train_rows": 5400,
    "raw_ece": 0.16046804825464883,
    "raw_brier": 0.05988674723912672,
    "raw_nll": 0.8413767837701986,
    "raw_accuracy": 0.7716666666666666,
    "raw_soft_accuracy": 0.5137750238180161,
    "raw_mean_confidence": 0.615313962996006,
    "raw_mean_target_max": 0.6725333379705747,
    "raw_score_mae": 0.24656671059007446,
    "raw_n": 600
  },
  {
    "epoch": 3,
    "step": 255,
    "loss": 0.8000329575373554,
    "sigma": 0.2,
    "epoch_seconds": 223.96768260002136,
    "train_rows": 5400,
    "raw_ece": 0.1488726359605789,
    "raw_brier": 0.05451012486705442,
    "raw_nll": 0.8328132717187205,
    "raw_accuracy": 0.78,
    "raw_soft_accuracy": 0.5247553873838235,
    "raw_mean_confidence": 0.6352111848195394,
    "raw_mean_target_max": 0.6725333379705747,
    "raw_score_mae": 0.22168685695699727,
    "raw_n": 600
  },
  {
    "epoch": 4,
    "step": 340,
    "loss": 0.7846576284726753,
    "sigma": 0.09999999999999998,
    "epoch_seconds": 222.44736337661743,
    "train_rows": 5400,
    "raw_ece": 0.13711758901675544,
    "raw_brier": 0.052573612098253716,
    "raw_nll": 0.8299004402291029,
    "raw_accuracy": 0.7766666666666666,
    "raw_soft_accuracy": 0.5293207626231015,
    "raw_mean_confidence": 0.6415973689158757,
    "raw_mean_target_max": 0.6725333379705747,
    "raw_score_mae": 0.21637691374926363,
    "raw_n": 600
  }
]