Text Classification
Transformers
PyTorch
English
roberta
distilroberta
sentiment
emotion
twitter
reddit
text-embeddings-inference
Instructions to use michelleli99/emotion_text_classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use michelleli99/emotion_text_classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="michelleli99/emotion_text_classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("michelleli99/emotion_text_classifier") model = AutoModelForSequenceClassification.from_pretrained("michelleli99/emotion_text_classifier") - Inference
- Notebooks
- Google Colab
- Kaggle
Michelle Li commited on
Commit ·
d3e4599
1
Parent(s): d080918
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ widget:
|
|
| 18 |
|
| 19 |
# Model Description
|
| 20 |
|
| 21 |
-
DistilRoBERTa-base is a transformer model that performs sentiment analysis. I fine-tuned the model on transcripts from the Friends show with the goal of classifying emotions from text data,
|
| 22 |
|
| 23 |
The model is a fine-tuned version of [Emotion English DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/) and [DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base). This model was initially trained on the following table (from [Emotion English DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/):
|
| 24 |
|
|
|
|
| 18 |
|
| 19 |
# Model Description
|
| 20 |
|
| 21 |
+
DistilRoBERTa-base is a transformer model that performs sentiment analysis. I fine-tuned the model on transcripts from the Friends show with the goal of classifying emotions from text data, specifically dialogue from Netflix shows or movies. The model predicts 6 Ekman emotions and a neutral class. These emotions include anger, disgust, fear, joy, neutrality, sadness, and surprise.
|
| 22 |
|
| 23 |
The model is a fine-tuned version of [Emotion English DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/) and [DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base). This model was initially trained on the following table (from [Emotion English DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/):
|
| 24 |
|