dair-ai/emotion
Viewer • Updated • 437k • 33.8k • 439
How to use Yanni8/bert-emotions with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Yanni8/bert-emotions") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Yanni8/bert-emotions")
model = AutoModelForSequenceClassification.from_pretrained("Yanni8/bert-emotions")This model is a fine-tuned version of bert-base-uncased on the emotion dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
Base model
google-bert/bert-base-uncased