Downloads · 30 days
6
13% of all-time downloads
t5_small_imdb
Minaaaa/t5_small_imdb
t5_small_imdb is a text classification model from Minaaaa. Use it when you need a label for a piece of text. It is set up for transformers.
T5-small is a fine-tuned version of T5-small model. The dataset used for fine tuning is the IMDB dataset from Hugging Face. The number of instances used for training is 12 500, and 2 500 for testing. Metrics used for…
All-time downloads
46
Public
Parameters
60.8M
243 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors243 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- t5
- Access
- Public
- Created
- Jan 11, 2024
- Updated
- Jan 17, 2024
- SHA
- 2373d6b6
Try a prompt
Datasets
Model card
From the Hugging Face model README
Model Card for Model ID
T5-small is a fine-tuned version of T5-small model. The dataset used for fine tuning is the IMDB dataset from Hugging Face. The number of instances used for training is 12 500, and 2 500 for testing. Metrics used for evaluation are classification oriented, because the model is used for sentiment analysis. The number of epochs used for training is 5, and the learning rate was set to lr = 5e-5. The results obtained are: Accuracy: 0.9204 Precision: 0.9158 Recall: 0.9268 F1-score: 0.9213
- Task
- Text Classification
- Library
- transformers
- Type
- t5
- Languages
- en
- Created
- Jan 11, 2024
- Updated
- Jan 17, 2024