Update README.md
Browse files
README.md
CHANGED
|
@@ -34,3 +34,13 @@ model = pipeline("text-classification", model="Kebinnuil/suicidal_detection_mode
|
|
| 34 |
|
| 35 |
result = model("I want to jump off the bridge")
|
| 36 |
print(result)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
result = model("I want to jump off the bridge")
|
| 36 |
print(result)
|
| 37 |
+
|
| 38 |
+
## Training Metrics
|
| 39 |
+
The dataset was split into 80/10/10 for train/validation/test set. Table below shows the result of the model's training metrics.
|
| 40 |
+
|
| 41 |
+
| Epoch | Training Loss | Validation Loss | Accuracy | AUC |
|
| 42 |
+
| :---- | :------------ | :-------------- | :------- | :------- |
|
| 43 |
+
| 1 | 0.442800 | 0.348061 | 0.838000 | 0.925000 |
|
| 44 |
+
| 2 | 0.304100 | 0.331631 | 0.850000 | 0.935000 |
|
| 45 |
+
| 3 | 0.261600 | 0.329701 | 0.851000 | 0.936000 |
|
| 46 |
+
|