End of training

README.md

@@ -10,22 +10,22 @@ metrics:
 - recall
 - f1
 model-index:
-- name: machiavellianism_binary
+- name: psychopathy_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. -->
 
-# machiavellianism_binary
+# psychopathy_binary
 
 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5728
+- Loss: 0.5869
 - Accuracy: 0.7047
-- Precision: 0.7690
-- Recall: 0.5054
-- F1: 0.6099
+- Precision: 0.7698
+- Recall: 0.4609
+- F1: 0.5766
 
 ## Model description
 
@@ -56,8 +56,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| No log        | 1.0   | 127  | 0.5810          | 0.7106   | 0.7584    | 0.5380 | 0.6294 |
-| No log        | 2.0   | 254  | 0.5728          | 0.7047   | 0.7690    | 0.5054 | 0.6099 |
+| No log        | 1.0   | 140  | 0.5715          | 0.6993   | 0.7254    | 0.5    | 0.5920 |
+| No log        | 2.0   | 280  | 0.5869          | 0.7047   | 0.7698    | 0.4609 | 0.5766 |
 
 
 ### Framework versions