Text Classification
Transformers
TensorBoard
Safetensors
distilbert
sentiment analysis
imdb
text-embeddings-inference
Instructions to use nppiech/an-imdb-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nppiech/an-imdb-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="nppiech/an-imdb-classifier")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("nppiech/an-imdb-classifier") model = AutoModelForSequenceClassification.from_pretrained("nppiech/an-imdb-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,8 @@ metrics:
|
|
| 9 |
model-index:
|
| 10 |
- name: an-imdb-classifier
|
| 11 |
results: []
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
@@ -60,4 +62,4 @@ The following hyperparameters were used during training:
|
|
| 60 |
- Transformers 4.55.0
|
| 61 |
- Pytorch 2.6.0+cu124
|
| 62 |
- Datasets 4.0.0
|
| 63 |
-
- Tokenizers 0.21.4
|
|
|
|
| 9 |
model-index:
|
| 10 |
- name: an-imdb-classifier
|
| 11 |
results: []
|
| 12 |
+
datasets:
|
| 13 |
+
- stanfordnlp/imdb
|
| 14 |
---
|
| 15 |
|
| 16 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
|
| 62 |
- Transformers 4.55.0
|
| 63 |
- Pytorch 2.6.0+cu124
|
| 64 |
- Datasets 4.0.0
|
| 65 |
+
- Tokenizers 0.21.4
|