Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
text-embeddings-inference
Instructions to use yuqinA/model_trained with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use yuqinA/model_trained with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="yuqinA/model_trained")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("yuqinA/model_trained") model = AutoModelForSequenceClassification.from_pretrained("yuqinA/model_trained", 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: model_trained
|
| 11 |
results: []
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
@@ -59,4 +61,4 @@ The following hyperparameters were used during training:
|
|
| 59 |
- Transformers 4.51.1
|
| 60 |
- Pytorch 2.6.0+cu124
|
| 61 |
- Datasets 3.5.0
|
| 62 |
-
- Tokenizers 0.21.1
|
|
|
|
| 9 |
model-index:
|
| 10 |
- name: model_trained
|
| 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
|
|
|
|
| 61 |
- Transformers 4.51.1
|
| 62 |
- Pytorch 2.6.0+cu124
|
| 63 |
- Datasets 3.5.0
|
| 64 |
+
- Tokenizers 0.21.1
|