|
--- |
|
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("Sakil/imdbsentdistilbertmodel") |
|
tokenizer = tokenizer.from_pretrained("Sakil/imdbsentdistilbertmodel") |
|
|
|
|