Downloads · 30 days
1.3K
0% of all-time downloads
joeddav/distilbert-base-uncased-go-emotions-student
distilbert-base-uncased-go-emotions-student is a text classification model from joeddav. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This model is distilled from the zero-shot classification pipeline on the unlabeled GoEmotions dataset using this script. It was trained with mixed precision for 10 epochs and otherwise used the default script arguments.
Downloads · 30 days
1.3K
0% of all-time downloads
All-time downloads
3.3M
Public
Parameters
67M
804 MB on disk
Likes
82
Public
Click a slice to open those files.
.h5268 MB · 33%
From the Hugging Face model README
This model is distilled from the zero-shot classification pipeline on the unlabeled GoEmotions dataset using this script. It was trained with mixed precision for 10 epochs and otherwise used the default script arguments.
The model can be used like any other model trained on GoEmotions, but will likely not perform as well as a model trained with full supervision. It is primarily intended as a demo of how an expensive NLI-based zero-shot model can be distilled to a more efficient student, allowing a classifier to be trained with only unlabeled data. Note that although the GoEmotions dataset allow multiple labels per instance, the teacher used single-label classification to create psuedo-labels.