| { |
| "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 |
| } |
| } |
