training_metadata.json

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