Text Classification
Transformers
TensorBoard
Safetensors
English
distilbert
sentiment
positive
negative
review
imdb
text-embeddings-inference
Instructions to use LukeGPT88/imdb_text_classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use LukeGPT88/imdb_text_classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="LukeGPT88/imdb_text_classifier")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("LukeGPT88/imdb_text_classifier") model = AutoModelForSequenceClassification.from_pretrained("LukeGPT88/imdb_text_classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -19,7 +19,8 @@ The model is a fine-tuned version of [DistilBERT](https://huggingface.co/docs/tr
|
|
| 19 |
|
| 20 |
It was fine-tuned on IMDB dataset [https://huggingface.co/datasets/imdb].
|
| 21 |
|
| 22 |
-
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on
|
|
|
|
| 23 |
It achieves the following results on the evaluation set:
|
| 24 |
- Loss: 0.2265
|
| 25 |
- Accuracy: 0.9312
|
|
|
|
| 19 |
|
| 20 |
It was fine-tuned on IMDB dataset [https://huggingface.co/datasets/imdb].
|
| 21 |
|
| 22 |
+
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on IMDB dataset.
|
| 23 |
+
|
| 24 |
It achieves the following results on the evaluation set:
|
| 25 |
- Loss: 0.2265
|
| 26 |
- Accuracy: 0.9312
|