Edit model card

Model Description

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!).

Training data

Training procedure

Preprocessing

Evaluation results

Limitations and bias

Well where should we begin...

EmpatheticDialogues:

  1. Unable to ascertain the degree of cultural specificity for the context that a respondent described when given an emotion label (i.e., p(description | emotion, culture))
  2. ...
Downloads last month
35
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train bdotloh/just-another-emotion-classifier