metadata
language: en
tags:
- text-classification
- tensorflow
- roberta
datasets:
- go_emotions
license: mit
What is the GoEmotions Dataset?
The dataset is comprised of 58000 Reddit comments with 28 emotions.
- 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
Usage
from transformers import RobertaTokenizerFast, TFRobertaForSequenceClassification, pipeline
tokenizer = RobertaTokenizerFast.from_pretrained("cappuch/EmoRoBERTa_Retrain")
model = TFRobertaForSequenceClassification.from_pretrained("cappuch/EmoRoBERTa_Retrain")
emotion = pipeline('sentiment-analysis',
model='cappuch/EmoRoBERTa_Retrain')
emotion_labels = emotion("Hello!")
print(emotion_labels)
#[{'label': 'neutral', 'score': 0.9964383244514465}]