Downloads · 30 days
27
7% of all-time downloads
sanjithrj/FeelWise
FeelWise is a feature extraction model from sanjithrj. Use it when you need embeddings to search or compare text. It is set up for transformers.
Downloads · 30 days
27
7% of all-time downloads
All-time downloads
364
Public
Parameters
13.6M
54.4 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors54.4 MB · 94%
From the Hugging Face model README
FeelWiseEmotion is a transformer-based emotion recognition model tailored for real-world applications that require nuanced understanding of human emotions. From detecting joy in social media posts to identifying distress in mental health messages, FeelWiseEmotion delivers semantically rich emotion classifications optimized for Large Language Models (LLMs) and downstream NLP tasks.
| Feature | Description |
|---|---|
| Transformer Encoder Backbone | Built using a custom multi-head self-attention encoder with AddNorm, FeedForward, and dropout. |
| Emotion-Aware Output | Classifies text into six universal emotions: joy, sadness, anger, fear, surprise, and disgust. |
| LLM-Optimized Representations | Mean pooling generates compact, interpretable vectors ideal for prompt engineering and fine-tuning. |
| Fully Configurable | Easily adjust depth, embedding dimensions, number of heads, FFN size, and vocab size via config. |
| Hugging Face Compatibility | Integrated with AutoModel and AutoTokenizer for frictionless usage in production pipelines. |
transformers library.pip install transformers torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model_name = "sanjithrj/FeelWiseEmotion"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
text = "I'm feeling really overwhelmed and anxious today."
inputs = tokenizer(text, return_tensors="pt")
logits = model(**inputs).logits
pred = torch.argmax(logits).item()
print(f"Predicted Emotion: {model.config.id2label[pred]}")
logits = model(**inputs).logits
# Output: tensor([[-0.18, 2.73, 0.41, 1.12, 0.09, -0.45]])
# → Predicted class index: 1 → Emotion: "Sadness"
{
"model_type": "FeelWiseEmotion",
"n_layers": 1,
"d_model": 256,
"n_head": 8,
"d_ff": 1024,
"max_len": 500,
"input_vocab_size": 50000,
"dropout": 0.1,
"num_classes": 6
}
Trained on the dair-ai/emotion dataset:
A popular benchmark for emotion classification covering multiple real-world expressions.
This model brings robust emotion understanding to the NLP pipeline with a lightweight yet extensible architecture—ideal for developers, researchers, and product teams building emotionally aware AI systems.
Would you like a badge for transformers, datasets, or model cards added as well?