File size: 930 Bytes
87e309d b477f51 87e309d b477f51 ab6b769 b477f51 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
---
license: mit
datasets:
- bhavyagiri/imdb-spoiler
language:
- en
metrics:
- accuracy
- f1
pipeline_tag: text-classification
tags:
- text-classification
- pytorch
- roberta
- emotions
- multi-class-classification
- multi-label-classification
widget:
- text: Jack Ryan is so amazing
---
The model trained from [roberta-base](https://huggingface.co/roberta-base) on the [imdb-spoiler](https://huggingface.co/datasets/bhavyagiri/imdb-spoiler) dataset for classification.
[imdb-spoiler](https://huggingface.co/datasets/bhavyagiri/imdb-spoiler) is a subset of a [large-dataset](https://www.kaggle.com/datasets/rmisra/imdb-spoiler-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.585
- Accuracy 0.474 |