End of training

README.md

@@ -9,22 +9,22 @@ metrics:
 - recall
 - f1
 model-index:
-- name: Neuro_binary
+- name: Imagination_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. -->
 
-# Neuro_binary
+# Imagination_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.5428
-- Accuracy: 0.7274
-- Precision: 0.7614
-- Recall: 0.7065
-- F1: 0.7329
+- Loss: 0.6389
+- Accuracy: 0.6278
+- Precision: 0.6697
+- Recall: 0.5338
+- F1: 0.5941
 
 ## Model description
 
@@ -55,8 +55,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| No log        | 1.0   | 135  | 0.5469          | 0.7265   | 0.7835    | 0.6678 | 0.7211 |
-| No log        | 2.0   | 270  | 0.5428          | 0.7274   | 0.7614    | 0.7065 | 0.7329 |
+| No log        | 1.0   | 134  | 0.6582          | 0.6082   | 0.5878    | 0.7770 | 0.6693 |
+| No log        | 2.0   | 268  | 0.6389          | 0.6278   | 0.6697    | 0.5338 | 0.5941 |
 
 
 ### Framework versions