antoniomenezes/go_emotions_ptbr
Updated • 69 • 6
How to use lukxus/GoEmotions with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="lukxus/GoEmotions") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("lukxus/GoEmotions")
model = AutoModelForSequenceClassification.from_pretrained("lukxus/GoEmotions")The following hyperparameters were used during training:
| Training Loss | Epoch | Validation Loss | F1 | F1 Macro |
|---|---|---|---|---|
| 1.238700 | 1.0 | 1.135360 | 0.238667 | 0.106812 |
| 1.084500 | 2.0 | 1.103713 | 0.234572 | 0.105525 |
| 1.009300 | 3.0 | 1.099275 | 0.290353 | 0.162856 |
| 0.929300 | 4.0 | 1.159663 | 0.334926 | 0.208453 |