google-research-datasets/go_emotions
Viewer • Updated • 265k • 12.1k • 264
How to use tuhanasinan/go_emotions-distilbert-tensorflow with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="tuhanasinan/go_emotions-distilbert-tensorflow", device_map="auto") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("tuhanasinan/go_emotions-distilbert-tensorflow")
model = AutoModelForSequenceClassification.from_pretrained("tuhanasinan/go_emotions-distilbert-tensorflow", device_map="auto")This model is a fine-tuned version of distilbert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Train Loss | Validation Loss | Train Accuracy | Epoch |
|---|---|---|---|
| 1.5400 | 1.3310 | 0.6010 | 0 |
| 1.1415 | 1.2910 | 0.6182 | 1 |
Base model
distilbert/distilbert-base-uncased