stanfordnlp/imdb
Viewer • Updated • 100k • 177k • 370
How to use jzonthemtn/distilbert-imdb with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="jzonthemtn/distilbert-imdb") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("jzonthemtn/distilbert-imdb")
model = AutoModelForSequenceClassification.from_pretrained("jzonthemtn/distilbert-imdb")This model is a fine-tuned version of distilbert-base-uncased on the imdb dataset.
Base model
distilbert/distilbert-base-uncased