File size: 351 Bytes
97bc050
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
---
license: mit
---
Model Distilbert fine-tuned with imdb reviews for sentiment analysis

from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast

model = DistilBertForSequenceClassification.from_pretrained("./distilbert-finetuned-imdb")
tokenizer = DistilBertTokenizerFast.from_pretrained("./distilbert-finetuned-imdb")