|
--- |
|
language: |
|
- en |
|
thumbnail: https://avatars3.githubusercontent.com/u/32437151?s=460&u=4ec59abc8d21d5feea3dab323d23a5860e6996a4&v=4 |
|
tags: |
|
- text-classification |
|
- emotion |
|
- pytorch |
|
- en |
|
- distil-bert |
|
license: apache-2.0 |
|
datasets: |
|
- emotion |
|
metrics: |
|
- Accuracy, F1 Score |
|
--- |
|
# Distilbert-base-uncased-emotion |
|
|
|
## Model description: |
|
`Distilbert-base-uncased` finetuned on the emotion dataset using HuggingFace Trainer. |
|
``` |
|
learning rate 2e-5, |
|
batch size 64, |
|
num_train_epochs=8, |
|
``` |
|
|
|
## Dataset: |
|
[Twitter-Sentiment-Analysis](https://huggingface.co/nlp/viewer/?dataset=emotion). |
|
|
|
#### Limitations and bias |
|
- |
|
- [Follow me on github](https://github.com/nateraw) to get notified when code is made available. |
|
|
|
## Training procedure |
|
|
|
## Eval results |
|
``` |
|
|
|
``` |
