How to use this model directly from the
from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("lvwerra/bert-imdb") model = AutoModelForSequenceClassification.from_pretrained("lvwerra/bert-imdb")
bert-large-cased) trained for sentiment classification on the IMDB dataset.
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.
The model achieved 90% classification accuracy on the validation set.
The full experiment is available in the tlr repo.