bhadresh-savani commited on
Commit
53ec0c6
1 Parent(s): f6532e6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -1
README.md CHANGED
@@ -15,13 +15,21 @@ metrics:
15
  # robert-base-emotion
16
 
17
  ## Model description:
18
- `robert-base` finetuned on the emotion dataset using HuggingFace Trainer.
19
  ```
20
  learning rate 2e-5,
21
  batch size 64,
22
  num_train_epochs=8,
23
  ```
24
 
 
 
 
 
 
 
 
 
25
  ## How to Use the model:
26
  ```python
27
  from transformers import pipeline
 
15
  # robert-base-emotion
16
 
17
  ## Model description:
18
+ `robert-base` finetuned on the emotion dataset using HuggingFace Trainer with below Hyperpatameters
19
  ```
20
  learning rate 2e-5,
21
  batch size 64,
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