dair-ai/emotion
Viewer • Updated • 437k • 23.6k • 449
How to use gokuls/hbertv1-emotion_48_emb_compress with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="gokuls/hbertv1-emotion_48_emb_compress") # Load model directly
from transformers import AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("gokuls/hbertv1-emotion_48_emb_compress", device_map="auto")This model is a fine-tuned version of gokuls/bert_12_layer_model_v1_complete_training_new_emb_compress_48 on the emotion 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:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 1.4218 | 1.0 | 250 | 1.1098 | 0.5885 |
| 0.9116 | 2.0 | 500 | 0.7865 | 0.743 |
| 0.5915 | 3.0 | 750 | 0.6149 | 0.805 |
| 0.4435 | 4.0 | 1000 | 0.4932 | 0.841 |
| 0.3626 | 5.0 | 1250 | 0.4634 | 0.855 |
| 0.3031 | 6.0 | 1500 | 0.4514 | 0.8545 |
| 0.2457 | 7.0 | 1750 | 0.4395 | 0.865 |
| 0.2039 | 8.0 | 2000 | 0.4368 | 0.861 |
| 0.1664 | 9.0 | 2250 | 0.4276 | 0.871 |
| 0.1402 | 10.0 | 2500 | 0.4493 | 0.874 |