Downloads · 30 days
11
7% of all-time downloads
emotion-minilm
Nikhil-iitj/emotion-minilm
emotion-minilm is a text classification model from Nikhil-iitj. Use it when you need a label for a piece of text. The card lists the license as mit.
This is the v1 model for our MLOps Group Project (Group 2, IIT Jodhpur).
All-time downloads
152
Public
Parameters
33.4M
801 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors133 MB · 99%
At a glance
- Task
- Text Classification
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Jun 10, 2026
- Updated
- Jun 10, 2026
- SHA
- 24a13779
Try a prompt
Base models
Model card
From the Hugging Face model README
Emotion Classification - MiniLM (v1)
This is the v1 model for our MLOps Group Project (Group 2, IIT Jodhpur).
Model Details
- Architecture: microsoft/MiniLM-L12-H384-uncased
- Task: Text emotion classification (6 classes)
- Dataset: dair-ai/emotion
- Classes: sadness, joy, love, anger, fear, surprise
Training
- Platform: Kaggle GPU T4
- Epochs: 3
- Batch size: 16
- Learning rate: 3e-5
- Test Accuracy: 90.7%
- Test F1: 90.7%
Links
- GitHub Repo: https://github.com/nikhilsaini-iitj/MLOps_GroupProject
- Kaggle Notebook: https://www.kaggle.com/code/nikhilg25ait2067/kaggle-minilm
- W&B Dashboard: https://wandb.ai/g25ait2067-prom-iit-rajasthan/mlops-groupproject-v2
- Winning Model (v2): https://huggingface.co/Nikhil-iitj/emotion-electra
Team
- Nikhil Saini (G25AIT2067)
- Y Sharathchandrika (G25AIT2132)
- Sarthak Kapoor (G25AIT2098)
- Aryaveer Rathi (G25AIT2021)
- Task
- Text Classification
- Type
- bert
- License
- mit
- Languages
- en
- Created
- Jun 10, 2026
- Updated
- Jun 10, 2026