model_info.json

{
  "model_name": "BERT Suicide Risk Detection",
  "model_type": "bert-base-uncased",
  "task": "Binary Classification (Suicide vs Non-Suicide)",
  "created_date": "2025-07-03T20:35:44.437244",
  "performance": {
    "test_f1": 0.9772,
    "test_accuracy": 0.9772,
    "test_precision": 0.9773,
    "test_recall": 0.9772,
    "best_val_f1": 0.9776
  },
  "training_details": {
    "epochs": 5,
    "batch_size": 32,
    "learning_rate": 2e-05,
    "max_length": 512,
    "dataset_size": 232074,
    "training_time": "approximately 45 minutes"
  },
  "usage_instructions": {
    "input": "Text string (any length)",
    "output": "Prediction (suicide/non-suicide) with confidence scores",
    "preprocessing": "Automatic text cleaning and tokenization"
  }
}