bhavyagiri
/

roberta-base-finetuned-imdb-spoilers

Text Classification

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

bhavyagiri commited on Aug 20, 2023

Commit

eb88fe8

•

1 Parent(s): 36946bf

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -26,8 +26,8 @@ The model trained from [roberta-base](https://huggingface.co/roberta-base) on th
 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
 Labels:
 - 0 - Not Spoiler

 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