dair-ai/emotion
Viewer • Updated • 437k • 24.5k • 449
How to use gokuls/hbertv2-emotion_48_emb_compress with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="gokuls/hbertv2-emotion_48_emb_compress") # Load model directly
from transformers import AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("gokuls/hbertv2-emotion_48_emb_compress", device_map="auto")This model is a fine-tuned version of gokuls/bert_12_layer_model_v2_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.5945 | 1.0 | 250 | 1.4950 | 0.478 |
| 1.352 | 2.0 | 500 | 1.1901 | 0.5595 |
| 1.0712 | 3.0 | 750 | 0.9287 | 0.651 |
| 0.8129 | 4.0 | 1000 | 0.7898 | 0.6955 |
| 0.6574 | 5.0 | 1250 | 0.7526 | 0.7335 |
| 0.5577 | 6.0 | 1500 | 0.6192 | 0.813 |
| 0.4418 | 7.0 | 1750 | 0.5638 | 0.8425 |
| 0.3931 | 8.0 | 2000 | 0.5432 | 0.8395 |
| 0.3536 | 9.0 | 2250 | 0.4958 | 0.8495 |
| 0.3184 | 10.0 | 2500 | 0.5104 | 0.851 |