README.md

---
license: apache-2.0
datasets:
- dair-ai/emotion
language:
- en
pipeline_tag: text-classification
---
## Sentiment Classifier: Emotion Detection with Sentence Transformer

This model classifies sentiments using the Sentence Transformer, specifically the `all-MiniLM-L6-v2` architecture. It's trained on the `dair-ai/emotion` dataset to identify six basic emotions:

* Sadness
* Joy
* Love
* Anger
* Fear
* Surprise

**Developed by:** shhossain: [https://github.com/shhossain](https://github.com/shhossain)

**Model type:** [Custom](https://huggingface.co/shhossain/all-MiniLM-L6-v2-sentiment-classifier/blob/main/model.py)

**Language(s):** English

**License:** Same as all-MiniLM-L6-v2

**Finetuned from:** all-MiniLM-L6-v2 ([https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2))

## Usage Example

```python
from transformers import pipeline

pipe = pipeline("text-classification", model="shhossain/all-MiniLM-L6-v2-sentiment-classifier", trust_remote_code=True)

result = pipe("This product is excellent!")
result
```

**Output:**

```json
[{'label': 'sad', 'score': 0.006396006792783737},
 {'label': 'joy', 'score': 0.7897642254829407},
 {'label': 'love', 'score': 0.17318710684776306},
 {'label': 'anger', 'score': 0.008878232911229134},
 {'label': 'fear', 'score': 0.010075093246996403},
 {'label': 'surprise', 'score': 0.011699344962835312}]
```