Commit
·
7037fa6
1
Parent(s):
e2b3525
Update README.md
Browse files
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/)
|