Downloads · 30 days
26
0% of all-time downloads
bert-imdb
lvwerra/bert-imdb
bert-imdb is a text classification model from lvwerra. Use it when you need a label for a piece of text. It is set up for transformers.
BERT (bert-large-cased) trained for sentiment classification on the IMDB dataset.
All-time downloads
31.9K
Public
Repo size
4 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin1.3 GB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- May 19, 2021
- SHA
- 2e60eb01
Try a prompt
Model card
From the Hugging Face model README
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.
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Mar 2, 2022
- Updated
- May 19, 2021