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:
- F1 0.773021
- Accuracy 0.783275
Labels:
- 0 - Not Spoiler
- 1 - Spoiler
- Downloads last month
- 6