Instructions to use lannelin/bert-imdb-1hidden with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use lannelin/bert-imdb-1hidden with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="lannelin/bert-imdb-1hidden")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("lannelin/bert-imdb-1hidden") model = AutoModelForSequenceClassification.from_pretrained("lannelin/bert-imdb-1hidden") - 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
Great, thanks!
lannelin changed pull request status to merged
You're welcome π€!