text-classification pytorch roberta imdb spoilers

The model trained from roberta-base on the imdb-spoiler dataset for classification.

imdb-spoiler is a subset of a large-dataset for classifying whether a movie review is a spoiler or not.

The model was trained using AutoModelForSequenceClassification.from_pretrained for 3 epochs with a learning rate of 2e-5 and weight decay of 0.01.

Evaluation using the dataset validation split gives:

Labels: