# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("aypan17/roberta-base-imdb")
model = AutoModelForSequenceClassification.from_pretrained("aypan17/roberta-base-imdb")Quick Links
TrainingArgs: lr=2e-5, train-batch-size=16, eval-batch-size=16, num-train-epochs=5, weight-decay=0.01,
- Downloads last month
- 3
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="aypan17/roberta-base-imdb")