Instructions to use XSY/albert-base-v2-imdb-calssification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use XSY/albert-base-v2-imdb-calssification with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="XSY/albert-base-v2-imdb-calssification")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("XSY/albert-base-v2-imdb-calssification") model = AutoModelForSequenceClassification.from_pretrained("XSY/albert-base-v2-imdb-calssification", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -26,6 +26,8 @@ model-index:
|
|
| 26 |
should probably proofread and complete it, then remove this comment. -->
|
| 27 |
|
| 28 |
# albert-base-v2-imdb-calssification
|
|
|
|
|
|
|
| 29 |
|
| 30 |
This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on the imdb dataset.
|
| 31 |
It achieves the following results on the evaluation set:
|
|
|
|
| 26 |
should probably proofread and complete it, then remove this comment. -->
|
| 27 |
|
| 28 |
# albert-base-v2-imdb-calssification
|
| 29 |
+
label_0: negative
|
| 30 |
+
label_1: positive
|
| 31 |
|
| 32 |
This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on the imdb dataset.
|
| 33 |
It achieves the following results on the evaluation set:
|