Downloads · 30 days
3
7% of all-time downloads
emotion-head-goemotions
Aharneish/emotion-head-goemotions
emotion-head-goemotions is a machine learning model from Aharneish. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository contains an auxiliary emotion classification head trained on GoEmotions, designed to be used with a frozen Mixture-of-Experts causal language model backbone.
All-time downloads
45
Public
Repo size
6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt6 MB · 100%
At a glance
- Access
- Public
- Created
- Jan 21, 2026
- Updated
- Jan 21, 2026
- SHA
- ca0c3bf2
Model card
From the Hugging Face model README
Emotion Classification Head (GoEmotions)
This repository contains an auxiliary emotion classification head trained on GoEmotions, designed to be used with a frozen Mixture-of-Experts causal language model backbone.
Architecture
- Input: final-token logits from backbone (
logits_to_keep=1) - Bottleneck projection: 512 dimensions
- Classifier: LayerNorm → Linear (28 labels)
Training
- Backbone: frozen
- Loss: CrossEntropy
- Precision: fp32 head, fp16 backbone
- Dataset: GoEmotions
Labels
admiration, amusement, anger, annoyance, approval, caring, confusion, curiosity, desire, disappointment, disapproval, disgust, embarrassment, excitement, fear, gratitude, grief, joy, love, nervousness, optimism, pride, realization, relief, remorse, sadness, surprise, neutral
Usage
Load emotion_head.pt and logit_bottleneck.pt and attach to a compatible backbone.
- Created
- Jan 21, 2026
- Updated
- Jan 21, 2026