Sakil
/

imdbsentdistilbertmodel

Text Classification

text Classification

Inference Endpoints

Model card Files Files and versions Community

Sakil commited on Jan 16, 2022

Commit

ba39f10

•

1 Parent(s): bed7d0f

Update README.md

Files changed (1) hide show

README.md +5 -0

README.md CHANGED Viewed

	@@ -13,3 +13,8 @@ widget:
13
14	- I have used IMDB movie review dataset to create custom model by using DistilBertForSequenceClassification.
15

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