stanfordnlp/imdb
Viewer • Updated • 100k • 269k • 379
How to use julien-c/distilbert-sagemaker-1609802168 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="julien-c/distilbert-sagemaker-1609802168") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("julien-c/distilbert-sagemaker-1609802168")
model = AutoModelForSequenceClassification.from_pretrained("julien-c/distilbert-sagemaker-1609802168")Trained from SageMaker HuggingFace extension.
Fine-tuned from distilbert-base-uncased on imdb 🔥
| key | value |
|---|---|
| eval_loss | 0.19187863171100616 |
| eval_accuracy | 0.9259 |
| eval_f1 | 0.9272173656811707 |
| eval_precision | 0.9147286821705426 |
| eval_recall | 0.9400517825134436 |
| epoch | 1.0 |