Downloads · 30 days
18
0% of all-time downloads
bdotloh/just-another-emotion-classifier
just-another-emotion-classifier is a text classification model from bdotloh. Use it when you need a label for a piece of text. It is set up for transformers.
Yet another Transformer model fine-tuned for approximating another non-linear mapping between X and Y? That's right! This is your good ol' emotion classifier - given an input text, the model outputs a probability dist…
Downloads · 30 days
18
0% of all-time downloads
All-time downloads
111K
Public
Repo size
876 MB
Likes
4
Public
Click a slice to open those files.
.bin438 MB · 100%
From the Hugging Face model README
Yet another Transformer model fine-tuned for approximating another non-linear mapping between X and Y? That's right! This is your good ol' emotion classifier - given an input text, the model outputs a probability distribution over a set of pre-selected emotion words. In this case, it is 32, which is the number of emotion classes in the Empathetic Dialogues dataset.
This model is built "on top of" a distilbert-base-uncased model fine-tuned on the go-emotions dataset. Y'all should really check out that model, it even contains a jupyter notebook file that illustrates how the model was trained (bhadresh-savani if you see this, thank you!).
Well where should we begin...
EmpatheticDialogues: