training_info.json

{
  "model": "training/best_model",
  "train_rows": 30967,
  "val_rows": 3441,
  "epochs_completed": 3.0,
  "best_f1": 0.8512281251502576,
  "training_time_hours": 0.2599656800429026,
  "class_weights": [
    1.0,
    2.8849579726508594
  ],
  "hyperparameters": {
    "max_length": 128,
    "batch_size": 16,
    "gradient_accumulation": 2,
    "learning_rate": 5e-06,
    "warmup_ratio": 0.1,
    "weight_decay": 0.01,
    "fp16": true,
    "seed": 42
  }
}