Downloads · 30 days
29
3% of all-time downloads
roberta-base-finetuned-imdb-spoilers
bhavyagiri/roberta-base-finetuned-imdb-spoilers
roberta-base-finetuned-imdb-spoilers is a text classification model from bhavyagiri. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
The model trained from roberta-base on the imdb-spoiler dataset for classification.
All-time downloads
1.1K
Public
Repo size
18.9 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin499 MB · 99%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- roberta
- Access
- Public
- Created
- Aug 18, 2023
- Updated
- Aug 20, 2023
- SHA
- eb88fe88
Try a prompt
Datasets
Model card
From the Hugging Face model README
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
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- License
- mit
- Languages
- en
- Created
- Aug 18, 2023
- Updated
- Aug 20, 2023