final_results.json

{
  "test_accuracy": 0.9772472636387142,
  "test_precision": 0.9772502424606044,
  "test_recall": 0.9772472636387142,
  "test_f1": 0.9772472268806155,
  "best_val_f1": 0.9776461055442124,
  "training_time": 1751574161.3237324,
  "config": {
    "model_name": "bert-base-uncased",
    "num_labels": 2,
    "max_length": 512,
    "batch_size": 32,
    "learning_rate": 2e-05,
    "num_epochs": 5,
    "warmup_steps": 0,
    "weight_decay": 0.01,
    "max_grad_norm": 1.0,
    "test_size": 0.2,
    "validation_size": 0.1,
    "use_mixed_precision": true,
    "gradient_accumulation_steps": 1,
    "logging_steps": 100,
    "save_steps": 500,
    "eval_steps": 500,
    "output_dir": "./bert_suicide_detection",
    "save_model": true
  }
}