Downloads · 30 days
35
2% of all-time downloads
sagemaker-roberta-base-emotion
Jorgeutd/sagemaker-roberta-base-emotion
sagemaker-roberta-base-emotion is a text classification model from Jorgeutd. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
This model is a fine-tuned model that was trained using Amazon SageMaker and the new Hugging Face Deep Learning container. - Problem type: Multi Class Text Classification (emotion detection).
All-time downloads
2K
Public
Parameters
125M
997 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin499 MB · 50%
Parameter types
How the weights are stored.
F32125M · 100%
Try a prompt
Datasets
Model card
From the Hugging Face model README
roberta-base
This model is a fine-tuned model that was trained using Amazon SageMaker and the new Hugging Face Deep Learning container.
- Problem type: Multi Class Text Classification (emotion detection).
It achieves the following results on the evaluation set:
- Loss: 0.1613253802061081
- f1: 0.9413321705151999
Hyperparameters
{
"epochs": 10,
"train_batch_size": 16,
"learning_rate": 3e-5,
"weight_decay":0.01,
"load_best_model_at_end": true,
"model_name":"roberta-base",
"do_eval": True,
"load_best_model_at_end":True
}
Validation Metrics
| key | value |
|---|---|
| eval_accuracy | 0.941 |
| eval_f1 | 0.9413321705151999 |
| eval_loss | 0.1613253802061081 |
| eval_recall | 0.941 |
| eval_precision | 0.9419519436781406 |
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- License
- apache-2.0
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- May 10, 2023