training_info.json

{
  "model_name": "microsoft/deberta-v3-base",
  "datasets_used": [
    "xTRam1/safe-guard-prompt-injection",
    "deepset/prompt-injections",
    "jayavibhav/prompt-injection-safety"
  ],
  "training_samples": 52903,
  "validation_samples": 5879,
  "final_metrics": {
    "eval_loss": 0.027365634217858315,
    "eval_accuracy": 0.995917673073652,
    "eval_precision": 0.9976027397260274,
    "eval_recall": 0.99419795221843,
    "eval_f1": 0.9958974358974358,
    "eval_runtime": 26.8342,
    "eval_samples_per_second": 219.086,
    "eval_steps_per_second": 27.39,
    "epoch": 3.0
  },
  "training_args": {
    "learning_rate": 3e-05,
    "batch_size": 8,
    "num_epochs": 3,
    "weight_decay": 0.01
  }
}