Update README.md

README.md

@@ -17,25 +17,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 # empathetic_dialogues_context_classification
 
-This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on an unknown dataset.
+This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on empatheticDialogues dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.6758
 - F1: 0.5315
 - Accuracy: 0.5315
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
+## Access to Code
+https://gist.github.com/zolfaShefreie/ae87ac2944e4f7b24609e0c28fde8449
 
 ### Training hyperparameters
 
@@ -62,6 +51,17 @@ The following hyperparameters were used during training:
 | 0.8075        | 8.0   | 4464 | 1.6758          | 0.5315 | 0.5315   |
 
 
+### Evaloation Result
+on EmpatheticDialogues dataset (test split) and using whole history as input:
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6503ebdce88eb2d0d51b33a8/NiJotkJsPDyihziT9gmz9.png)
+
+on [empathetic-dialogues-contexts](https://huggingface.co/datasets/bdotloh/empathetic-dialogues-contexts) dataset (test split):
+ 
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6503ebdce88eb2d0d51b33a8/QusXIpjz0A0vf_R77szMu.png)
+
+
 ### Framework versions
 
 - Transformers 4.44.2