finetuning complete

README.md

@@ -4,22 +4,18 @@ base_model: roberta-base
 tags:
 - generated_from_keras_callback
 model-index:
-- name: rasyosef/roberta-base-finetuned-sst2
+- name: roberta-base-finetuned-sst2
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
 probably proofread and complete it, then remove this comment. -->
 
-# rasyosef/roberta-base-finetuned-sst2
+# roberta-base-finetuned-sst2
 
 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0760
-- Train Accuracy: 0.9736
-- Validation Loss: 0.2081
-- Validation Accuracy: 0.9346
-- Epoch: 2
+
 
 ## Model description
 
@@ -43,11 +39,6 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
-|:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
-| 0.2126     | 0.9165         | 0.1748          | 0.9335              | 0     |
-| 0.1223     | 0.9562         | 0.1806          | 0.9381              | 1     |
-| 0.0760     | 0.9736         | 0.2081          | 0.9346              | 2     |
 
 
 ### Framework versions

config.json

@@ -10,8 +10,16 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
+  "id2label": {
+    "0": "negative",
+    "1": "positive"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
+  "label2id": {
+    "negative": "0",
+    "positive": "1"
+  },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
   "model_type": "roberta",