sinensia's picture
Update README.md
97bc050 verified
|
raw
history blame
No virus
351 Bytes
metadata
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")