Edit model card

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

Downloads last month
165
Safetensors
Model size
60.8M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train Minaaaa/t5_small_imdb