End of training

README.md

@@ -1,6 +1,6 @@
 ---
 license: mit
-base_model: roberta-base
+base_model: roberta-large
 tags:
 - generated_from_trainer
 metrics:
@@ -18,13 +18,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Imagination_binary
 
-This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
+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.6389
-- Accuracy: 0.6278
-- Precision: 0.6697
-- Recall: 0.5338
-- F1: 0.5941
+- Loss: 0.6415
+- Accuracy: 0.6446
+- Precision: 0.6859
+- Recall: 0.5161
+- F1: 0.5890
+- Auc: 0.6429
 
 ## Model description
 
@@ -53,10 +54,10 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| 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 |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     | Auc    |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:------:|
+| No log        | 1.0   | 134  | 0.6696          | 0.5756   | 0.5467    | 0.8185 | 0.6556 | 0.5787 |
+| No log        | 2.0   | 268  | 0.6415          | 0.6446   | 0.6859    | 0.5161 | 0.5890 | 0.6429 |
 
 
 ### Framework versions