PrajwalNayaka/Text-Emotion
Viewer • Updated • 106k • 50
How to use auliganteng/my_distilbert_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="auliganteng/my_distilbert_model") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("auliganteng/my_distilbert_model")
model = AutoModelForSequenceClassification.from_pretrained("auliganteng/my_distilbert_model")This model is a fine-tuned version of distilbert-base-uncased on the None 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 | F1 | Precision | Recall |
|---|---|---|---|---|---|---|---|
| 0.0974 | 1.0 | 20841 | 0.0939 | 0.9398 | 0.9398 | 0.9469 | 0.9398 |
| 0.0838 | 2.0 | 41682 | 0.0935 | 0.9419 | 0.9428 | 0.9494 | 0.9419 |
| 0.0804 | 3.0 | 62523 | 0.1010 | 0.9410 | 0.9422 | 0.9475 | 0.9410 |
Base model
distilbert/distilbert-base-uncased