Shortcuts

Source code for torchexpo.nlp.sentiment_analysis.distilbert

import transformers
from torchexpo.modules import SentimentAnalysisModule


[docs]def distilbert_imdb(): """DistilBERT Model pre-trained on IMDb""" model = transformers.DistilBertForSequenceClassification.from_pretrained( "textattack/distilbert-base-uncased-imdb", torchscript=True) obj = SentimentAnalysisModule(model, "DistilBERT IMDb", model_example="default") return obj
Read the Docs v: latest
Versions
latest
Downloads
pdf
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.