bhadresh-savani
commited on
Commit
·
53ec0c6
1
Parent(s):
f6532e6
Update README.md
Browse files
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
|