bhadresh-savani commited on
Commit
7037fa6
1 Parent(s): e2b3525

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -6
README.md CHANGED
@@ -22,6 +22,14 @@ metrics:
22
  num_train_epochs=8,
23
  ```
24
 
 
 
 
 
 
 
 
 
25
  ## How to Use the model:
26
  ```python
27
  from transformers import pipeline
@@ -63,11 +71,5 @@ Output:
63
  }
64
  ```
65
 
66
- ## Model Performance Comparision on Emotion Dataset from Twitter:
67
-
68
- | Model | Accuracy | F1 Score | Test Loss | Test Sample per Second |
69
- | --- | --- | --- | --- | --- |
70
- | Distilbert | 93.8 | 93.79 | 0.1472 | 398.69 |
71
-
72
  ## Reference:
73
  * [Natural Language Processing with Transformer By Lewis Tunstall, Leandro von Werra, Thomas Wolf](https://learning.oreilly.com/library/view/natural-language-processing/9781098103231/)
 
22
  num_train_epochs=8,
23
  ```
24
 
25
+ ## Model Performance Comparision on Emotion Dataset from Twitter:
26
+
27
+ | Model | Accuracy | F1 Score | Test Sample per Second |
28
+ | --- | --- | --- | --- |
29
+ | Distilbert-base-uncased | 93.8 | 93.79 | 398.69 |
30
+ | Bert-base-uncased | 94.05 | 94.06 | 190.152 |
31
+ | Roberta-base | 93.95 | 93.97| 195.639 |
32
+
33
  ## How to Use the model:
34
  ```python
35
  from transformers import pipeline
 
71
  }
72
  ```
73
 
 
 
 
 
 
 
74
  ## Reference:
75
  * [Natural Language Processing with Transformer By Lewis Tunstall, Leandro von Werra, Thomas Wolf](https://learning.oreilly.com/library/view/natural-language-processing/9781098103231/)