End of training

README.md

@@ -10,22 +10,22 @@ metrics:
 - recall
 - f1
 model-index:
-- name: Anger_binary
+- name: Depression_binary
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# Anger_binary
+# Depression_binary
 
 This model is a fine-tuned version of [roberta-large](https://huggingface.co/roberta-large) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6933
-- Accuracy: 0.6381
-- Precision: 0.6248
-- Recall: 0.6379
-- F1: 0.6313
+- Loss: 0.5524
+- Accuracy: 0.7613
+- Precision: 0.7254
+- Recall: 0.7076
+- F1: 0.7164
 
 ## Model description
 
@@ -56,9 +56,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| No log        | 1.0   | 135  | 0.6440          | 0.6307   | 0.6513    | 0.5153 | 0.5754 |
-| No log        | 2.0   | 270  | 0.6462          | 0.6540   | 0.6410    | 0.6533 | 0.6471 |
-| No log        | 3.0   | 405  | 0.6933          | 0.6381   | 0.6248    | 0.6379 | 0.6313 |
+| No log        | 1.0   | 144  | 0.5393          | 0.7509   | 0.7636    | 0.6012 | 0.6728 |
+| No log        | 2.0   | 288  | 0.5035          | 0.7544   | 0.7820    | 0.5869 | 0.6706 |
+| No log        | 3.0   | 432  | 0.5524          | 0.7613   | 0.7254    | 0.7076 | 0.7164 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d9593e7816b177f4ebed2e194e0b4908ff22c55b5467c6223a2f618d00399bdb
+oid sha256:44806e314a6d07ffdbbea00f27348fc6c4d02673c645c1e23e6d41e508ac9670
 size 1421495416