Upload model_info.json

model_info.json

@@ -0,0 +1,26 @@
+{
+  "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"
+  }
+}