Text Classification
Transformers
PyTorch
distilbert
Generated from Trainer
text-embeddings-inference
Instructions to use lkm2835/distilbert-imdb with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use lkm2835/distilbert-imdb with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="lkm2835/distilbert-imdb")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("lkm2835/distilbert-imdb") model = AutoModelForSequenceClassification.from_pretrained("lkm2835/distilbert-imdb") - Notebooks
- Google Colab
- Kaggle
Align label mapping with imdb dataset
#1
by lewtun HF Staff - opened
Hi there, your model is using a default label mapping. Accept this PR to align the label mapping with the imdb dataset this model was trained on. This will enable your model to be evaluated by Hugging Face's automatic model evaluator
Thanks!
lkm2835 changed pull request status to merged