240 lines
6.1 KiB
JSON
240 lines
6.1 KiB
JSON
{
|
|
"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
|
|
}
|
|
]
|
|
} |