NLP Roberta Version! 🤗

README.md

@@ -2,6 +2,7 @@
 license: apache-2.0
 base_model: distilroberta-base
 tags:
+- text-classification
 - generated_from_trainer
 datasets:
 - glue
@@ -15,7 +16,7 @@ model-index:
       name: Text Classification
       type: text-classification
     dataset:
-      name: glue
+      name: datasetX
       type: glue
       config: mrpc
       split: validation
@@ -23,10 +24,10 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.8137254901960784
+      value: 0.7965686274509803
     - name: F1
       type: f1
-      value: 0.8685121107266436
+      value: 0.8482632541133455
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,11 +35,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # platzi_nlp_model_roberta_similaritytext
 
-This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the glue dataset.
+This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the datasetX dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.3339
-- Accuracy: 0.8137
-- F1: 0.8685
+- Loss: 0.9276
+- Accuracy: 0.7966
+- F1: 0.8483
 
 ## Model description
 

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 5.0,
+    "eval_accuracy": 0.7965686274509803,
+    "eval_f1": 0.8482632541133455,
+    "eval_loss": 0.9275822639465332,
+    "eval_runtime": 1.4492,
+    "eval_samples_per_second": 281.543,
+    "eval_steps_per_second": 35.193,
+    "train_loss": 0.13423540535056253,
+    "train_runtime": 158.7008,
+    "train_samples_per_second": 115.563,
+    "train_steps_per_second": 14.461
+}

eval_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 5.0,
+    "eval_accuracy": 0.7965686274509803,
+    "eval_f1": 0.8482632541133455,
+    "eval_loss": 0.9275822639465332,
+    "eval_runtime": 1.4492,
+    "eval_samples_per_second": 281.543,
+    "eval_steps_per_second": 35.193
+}

train_results.json

@@ -0,0 +1,7 @@
+{
+    "epoch": 5.0,
+    "train_loss": 0.13423540535056253,
+    "train_runtime": 158.7008,
+    "train_samples_per_second": 115.563,
+    "train_steps_per_second": 14.461
+}