Update README.md

README.md

@@ -1,5 +1,5 @@
 ---
-license: apache-2.0
+license: mit
 datasets:
 - bhavyagiri/imdb-spoiler
 language:
@@ -9,5 +9,23 @@ metrics:
 - f1
 pipeline_tag: text-classification
 tags:
-- code
----
+- 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 (of the 28 dim output via a threshold of 0.5 to binarize each) using the dataset test split gives:
+- F1 0.585
+- Accuracy 0.474