Shortcuts

Source code for torchexpo.nlp.sentiment_analysis.electra

import transformers
from torchexpo.modules import SentimentAnalysisModule


[docs]def electra_imdb(): """Electra Model pre-trained on IMDb""" model = transformers.ElectraForSequenceClassification.from_pretrained( "monologg/electra-small-finetuned-imdb", torchscript=True) obj = SentimentAnalysisModule(model, "Electra 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.