training_summary.json

{
  "model_name": "distilbert-base-uncased",
  "num_epochs": 15,
  "batch_size": 16,
  "learning_rate": 3e-05,
  "train_samples": 659,
  "val_samples": 141,
  "final_metrics": {
    "eval_loss": 0.32609713077545166,
    "eval_accuracy": 0.9290780141843972,
    "eval_f1": 0.9268291188191063,
    "eval_precision": 0.9317931083888531,
    "eval_recall": 0.9290780141843972,
    "eval_runtime": 0.4923,
    "eval_samples_per_second": 286.387,
    "eval_steps_per_second": 18.28,
    "epoch": 10.0
  },
  "total_params": 66625416,
  "trainable_params": 66625416
}