{
  "mode": "v0.0.2",
  "backend": "pytorch-cuda",
  "epochs_trained": 29,
  "train_time_s": 0.7185604920377955,
  "best_eval_acc": 0.6232557892799377,
  "final_eval_acc": 0.6232557892799377,
  "final_eval_within_pm1": 1.0,
  "final_eval_mae": 0.37674418091773987,
  "temperature_scale": 0.9261822700500488,
  "conf_correctness_spearman_post_temp": 0.012770170735830375,
  "per_class_accuracy": {
    "0": {
      "support": 116,
      "accuracy": 0.8620689655172413
    },
    "1": {
      "support": 99,
      "accuracy": 0.3434343434343434
    }
  },
  "hyperparameters": {
    "optimizer": "AdamW",
    "lr": 0.001,
    "weight_decay": 0.01,
    "batch_size": 64,
    "schedule": "cosine_warm_restarts",
    "epochs_max": 400,
    "label_smoothing": 0.1,
    "patience": 20,
    "split": "random_80_20_seed_42",
    "balanced_sampler": true,
    "temperature_scaling": true
  },
  "epoch_losses": [
    {
      "epoch": 0,
      "train_loss": 1.8680313183711126,
      "train_acc": 0.4543269230769231,
      "eval_loss": 0.7276814579963684,
      "eval_acc": 0.539534866809845
    },
    {
      "epoch": 1,
      "train_loss": 1.3579198305423443,
      "train_acc": 0.5060096153846154,
      "eval_loss": 0.8614012002944946,
      "eval_acc": 0.46046510338783264
    },
    {
      "epoch": 2,
      "train_loss": 1.299364447593689,
      "train_acc": 0.4831730769230769,
      "eval_loss": 0.7327257990837097,
      "eval_acc": 0.539534866809845
    },
    {
      "epoch": 3,
      "train_loss": 1.2834151433064387,
      "train_acc": 0.4963942307692308,
      "eval_loss": 0.7958587408065796,
      "eval_acc": 0.539534866809845
    },
    {
      "epoch": 4,
      "train_loss": 1.2809640077444224,
      "train_acc": 0.49278846153846156,
      "eval_loss": 0.7728011608123779,
      "eval_acc": 0.46046510338783264
    },
    {
      "epoch": 5,
      "train_loss": 1.276416512636038,
      "train_acc": 0.5120192307692307,
      "eval_loss": 0.7620130181312561,
      "eval_acc": 0.539534866809845
    },
    {
      "epoch": 6,
      "train_loss": 1.2767094740500817,
      "train_acc": 0.4951923076923077,
      "eval_loss": 0.7696149945259094,
      "eval_acc": 0.604651153087616
    },
    {
      "epoch": 7,
      "train_loss": 1.2724562699978168,
      "train_acc": 0.5324519230769231,
      "eval_loss": 0.7653729319572449,
      "eval_acc": 0.539534866809845
    },
    {
      "epoch": 8,
      "train_loss": 1.2739891455723689,
      "train_acc": 0.5264423076923077,
      "eval_loss": 0.7635467648506165,
      "eval_acc": 0.6232557892799377
    },
    {
      "epoch": 9,
      "train_loss": 1.2718101739883423,
      "train_acc": 0.5120192307692307,
      "eval_loss": 0.7564782500267029,
      "eval_acc": 0.604651153087616
    },
    {
      "epoch": 10,
      "train_loss": 1.261798886152414,
      "train_acc": 0.5625,
      "eval_loss": 0.7915780544281006,
      "eval_acc": 0.46046510338783264
    },
    {
      "epoch": 11,
      "train_loss": 1.2723550613109882,
      "train_acc": 0.5348557692307693,
      "eval_loss": 0.7585318088531494,
      "eval_acc": 0.6139534711837769
    },
    {
      "epoch": 12,
      "train_loss": 1.2408426174750695,
      "train_acc": 0.6225961538461539,
      "eval_loss": 0.7562077045440674,
      "eval_acc": 0.525581419467926
    },
    {
      "epoch": 13,
      "train_loss": 1.219417168543889,
      "train_acc": 0.6334134615384616,
      "eval_loss": 0.7647078633308411,
      "eval_acc": 0.5860465168952942
    },
    {
      "epoch": 14,
      "train_loss": 1.198713256762578,
      "train_acc": 0.6526442307692307,
      "eval_loss": 0.7711634635925293,
      "eval_acc": 0.5720930099487305
    },
    {
      "epoch": 15,
      "train_loss": 1.167367669252249,
      "train_acc": 0.6826923076923077,
      "eval_loss": 0.7664391994476318,
      "eval_acc": 0.6186046600341797
    },
    {
      "epoch": 16,
      "train_loss": 1.1867470557873065,
      "train_acc": 0.6574519230769231,
      "eval_loss": 0.7853891253471375,
      "eval_acc": 0.6139534711837769
    },
    {
      "epoch": 17,
      "train_loss": 1.185251813668471,
      "train_acc": 0.6766826923076923,
      "eval_loss": 0.7728492021560669,
      "eval_acc": 0.5767441987991333
    },
    {
      "epoch": 18,
      "train_loss": 1.1749065747627845,
      "train_acc": 0.6814903846153846,
      "eval_loss": 0.7930512428283691,
      "eval_acc": 0.5488371849060059
    },
    {
      "epoch": 19,
      "train_loss": 1.1521984338760376,
      "train_acc": 0.6983173076923077,
      "eval_loss": 0.7875214219093323,
      "eval_acc": 0.5860465168952942
    },
    {
      "epoch": 20,
      "train_loss": 1.158121026479281,
      "train_acc": 0.6802884615384616,
      "eval_loss": 0.785778820514679,
      "eval_acc": 0.5860465168952942
    },
    {
      "epoch": 21,
      "train_loss": 1.1232389486753023,
      "train_acc": 0.7319711538461539,
      "eval_loss": 0.7949181795120239,
      "eval_acc": 0.5767441987991333
    },
    {
      "epoch": 22,
      "train_loss": 1.1163162634922907,
      "train_acc": 0.7391826923076923,
      "eval_loss": 0.867073118686676,
      "eval_acc": 0.539534866809845
    },
    {
      "epoch": 23,
      "train_loss": 1.1119057948772724,
      "train_acc": 0.7211538461538461,
      "eval_loss": 0.8135209679603577,
      "eval_acc": 0.5953488349914551
    },
    {
      "epoch": 24,
      "train_loss": 1.107274578167842,
      "train_acc": 0.7271634615384616,
      "eval_loss": 0.8401668071746826,
      "eval_acc": 0.5534883737564087
    },
    {
      "epoch": 25,
      "train_loss": 1.0781027399576628,
      "train_acc": 0.7451923076923077,
      "eval_loss": 0.8606341481208801,
      "eval_acc": 0.5441860556602478
    },
    {
      "epoch": 26,
      "train_loss": 1.041811819259937,
      "train_acc": 0.7584134615384616,
      "eval_loss": 0.8801625967025757,
      "eval_acc": 0.5767441987991333
    },
    {
      "epoch": 27,
      "train_loss": 1.0369769976689265,
      "train_acc": 0.7764423076923077,
      "eval_loss": 0.8642652034759521,
      "eval_acc": 0.5860465168952942
    },
    {
      "epoch": 28,
      "train_loss": 1.0502384350850031,
      "train_acc": 0.7524038461538461,
      "eval_loss": 0.8719286322593689,
      "eval_acc": 0.5720930099487305
    }
  ]
}