Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/models-cards#model-card-metadata)

BERT-IMDB

What is it?

BERT (bert-large-cased) trained for sentiment classification on the IMDB dataset.

Training setting

The model was trained on 80% of the IMDB dataset for sentiment classification for three epochs with a learning rate of 1e-5 with the simpletransformers library. The library uses a learning rate schedule.

Result

The model achieved 90% classification accuracy on the validation set.

Reference

The full experiment is available in the tlr repo.

Downloads last month
1,012
Hosted inference API
Text Classification
Examples
Examples
This model can be loaded on the Inference API on-demand.