Upload final_results.json

final_results.json

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