Instructions to use Den953/distilbert-imdb-workshop with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Den953/distilbert-imdb-workshop with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Den953/distilbert-imdb-workshop")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Den953/distilbert-imdb-workshop") model = AutoModelForSequenceClassification.from_pretrained("Den953/distilbert-imdb-workshop", device_map="auto") - Notebooks
- Google Colab
- Kaggle
distilbert-imdb-workshop
distilbert-base-uncased fine-tune sur un sous-echantillon d'
imdb pour la classification
binaire de sentiment.
Atelier Get_Started du cours MLOps M2. Evalue avec accuracy et F1, les deux
metriques demandees par l'enonce.
Resultats
| Metrique | Valeur |
|---|---|
| Accuracy | 0.8275 |
| F1 | 0.8208 |
L'accuracy est la proportion de predictions correctes ; le F1 est la moyenne harmonique de la precision et du rappel. Les deux sont proches ici parce que les classes d'IMDB sont equilibrees.
Utilisation
from transformers import pipeline
clf = pipeline("sentiment-analysis", model="Den953/distilbert-imdb-workshop", truncation=True)
clf("This movie was absolutely fantastic!")
Entrainement
| Hyperparametre | Valeur |
|---|---|
| base | distilbert-base-uncased |
| train / test | 1500 / 400 exemples |
| epochs | 1 |
| learning rate | 2e-5 |
| batch size | 16 |
| weight decay | 0.01 |
| max length | 128 tokens (padding dynamique) |
| materiel | CPU |
Limites
Entrainement volontairement court (1500 exemples, 1 epoque, CPU) : exercice pedagogique, pas un modele de production. Avec 400 exemples de test, l'accuracy rapportee a une marge d'erreur de l'ordre de +/- 5 points. Critiques de films en anglais uniquement, souvent longues et tronquees a 128 tokens - le modele ne voit donc que le debut de chaque critique.
- Downloads last month
- 20
Model tree for Den953/distilbert-imdb-workshop
Base model
distilbert/distilbert-base-uncasedDataset used to train Den953/distilbert-imdb-workshop
Evaluation results
- Accuracy on imdbtest set self-reported0.828
- F1 on imdbtest set self-reported0.821