Text Classification
Transformers
TensorBoard
ONNX
Safetensors
PyTorch
English
distilbert
movie-review-sentiment
BertForSequenceClassification
Generated from Trainer
text-embeddings-inference
Instructions to use pitangent-ds/distilbert-base-imdb with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use pitangent-ds/distilbert-base-imdb with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="pitangent-ds/distilbert-base-imdb")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("pitangent-ds/distilbert-base-imdb") model = AutoModelForSequenceClassification.from_pretrained("pitangent-ds/distilbert-base-imdb") - Notebooks
- Google Colab
- Kaggle
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.9315, | |
| "eval_loss": 0.34898611903190613, | |
| "eval_matthews_correlation": 0.8630000566256675, | |
| "eval_runtime": 51.1853, | |
| "eval_samples_per_second": 195.369, | |
| "eval_steps_per_second": 0.625 | |
| } |