language: | |
- en | |
tags: | |
- text Classification | |
license: apache-2.0 | |
widget: | |
- text: "I like you. </s></s> I love you." | |
* IMDBSentimentDistilBertModel: | |
- I have used IMDB movie review dataset to create custom model by using DistilBertForSequenceClassification. | |
from transformers import DistilBertForSequenceClassification, Trainer, TrainingArguments | |
model = DistilBertForSequenceClassification.from_pretrained('./imdbsentdistilbertmodel') | |