metadata.json

{
  "model_name": "WIP2",
  "base_model": "meta-llama/Llama-Prompt-Guard-2-22M",
  "fine_tuning_method": "LoRA",
  "dataset": "qualifire/prompt-injections-benchmark",
  "num_labels": 2,
  "lora_rank": 16,
  "lora_alpha": 32,
  "lora_dropout": 0.1,
  "max_length": 512,
  "final_metrics": {
    "eval_loss": 0.2866095304489136,
    "eval_accuracy": 0.9197080291970803,
    "eval_f1": 0.8929068150208623,
    "eval_precision": 0.9525222551928784,
    "eval_recall": 0.8403141361256544,
    "eval_roc_auc": 0.9797109076555935,
    "eval_runtime": 18.0185,
    "eval_samples_per_second": 53.223,
    "eval_steps_per_second": 6.66,
    "epoch": 5.0
  },
  "hf_ready": true,
  "merge_lora": true
}