training_config.json

{
  "model_name": "pminervini/RoBERTa-base-PM-M3-Voc-hf",
  "max_length": 512,
  "batch_size": 32,
  "learning_rate": 2e-05,
  "num_epochs": 3,
  "test_size": 0.2,
  "random_state": 42,
  "training_samples": 70237,
  "validation_samples": 17560,
  "final_metrics": {
    "eval_loss": 0.5086122751235962,
    "eval_accuracy": 0.767369020501139,
    "eval_precision": 0.7358,
    "eval_recall": 0.8360413589364845,
    "eval_f1": 0.7827243231742992,
    "eval_runtime": 20.6312,
    "eval_samples_per_second": 851.138,
    "eval_steps_per_second": 13.329,
    "epoch": 3.0
  }
}