---
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')