How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="bergr7/imdb-classification-dl-enero-final")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("bergr7/imdb-classification-dl-enero-final")
model = AutoModelForSequenceClassification.from_pretrained("bergr7/imdb-classification-dl-enero-final", device_map="auto")
Quick Links

YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

IMDB Classification with BERT


license: mit datasets: - imdb language: - en metrics: - accuracy - f1 library_name: transformers pipeline_tag: text-classification

Downloads last month
3
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support