Migrate model card from transformers-repo

3c84a28 almost 6 years ago

619 Bytes

# BERT-IMDB

## What is it?
BERT (`bert-large-cased`) trained for sentiment classification on the [IMDB dataset](https://www.kaggle.com/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews).

## 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](https://lvwerra.github.io/trl/03-bert-imdb-training/).

	# BERT-IMDB

	## What is it?
	BERT (`bert-large-cased`) trained for sentiment classification on the [IMDB dataset](https://www.kaggle.com/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews).

	## 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](https://lvwerra.github.io/trl/03-bert-imdb-training/).