Glepka/kinopoisk_classification
Viewer • Updated • 58.5k • 9
How to use Glepka/rubert-tiny2-kinopoisk with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Glepka/rubert-tiny2-kinopoisk") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Glepka/rubert-tiny2-kinopoisk")
model = AutoModelForSequenceClassification.from_pretrained("Glepka/rubert-tiny2-kinopoisk", device_map="auto")This model is a fine-tuned version of cointegrated/rubert-tiny2 on Glepka/kinopoisk_classification. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.7896 | 1.0 | 891 | 0.5692 | 0.7367 |
| 0.5548 | 2.0 | 1782 | 0.5882 | 0.734 |
| 0.5207 | 3.0 | 2673 | 0.5731 | 0.7467 |
| 0.4806 | 4.0 | 3564 | 0.5806 | 0.7487 |
| 0.4615 | 5.0 | 4455 | 0.6030 | 0.746 |
| 0.4206 | 6.0 | 5346 | 0.6197 | 0.7453 |
| 0.3913 | 7.0 | 6237 | 0.6427 | 0.7413 |
| 0.3678 | 8.0 | 7128 | 0.6605 | 0.7413 |
| 0.3392 | 9.0 | 8019 | 0.6922 | 0.7367 |
| 0.3299 | 10.0 | 8910 | 0.7000 | 0.738 |
| 0.3125 | 11.0 | 9801 | 0.7139 | 0.736 |
| 0.308 | 12.0 | 10692 | 0.7214 | 0.738 |
Base model
cointegrated/rubert-tiny2