File size: 1,207 Bytes
7553492 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
{
"model_name": "beomi/KcELECTRA-base-v2022",
"task": "binary_text_classification",
"num_labels": 2,
"training_args": {
"num_train_epochs": 5,
"per_device_train_batch_size": 32,
"per_device_eval_batch_size": 64,
"learning_rate": 3e-05,
"weight_decay": 0.01,
"warmup_ratio": 0.1,
"bf16": true,
"optimizer": "adamw_torch_fused",
"seed": 42
},
"training_time_minutes": 1.87,
"validation_metrics": {
"eval_loss": 0.06963474303483963,
"eval_accuracy": 0.9757751937984496,
"eval_f1": 0.9480249480249481,
"eval_precision": 0.9382716049382716,
"eval_recall": 0.957983193277311,
"eval_runtime": 0.5253,
"eval_samples_per_second": 1964.52,
"eval_steps_per_second": 32.361,
"epoch": 3.409090909090909
},
"test_metrics": {
"eval_loss": 0.26781579852104187,
"eval_accuracy": 0.8930817610062893,
"eval_f1": 0.826530612244898,
"eval_precision": 0.9204545454545454,
"eval_recall": 0.75,
"eval_runtime": 0.1474,
"eval_samples_per_second": 2156.782,
"eval_steps_per_second": 33.912,
"epoch": 3.409090909090909
},
"confusion_matrix": {
"TN": 203,
"FP": 7,
"FN": 27,
"TP": 81
}
} |