|
--- |
|
license: mit |
|
language: |
|
- ru |
|
pipeline_tag: text-classification |
|
tags: |
|
- sentiment-analysis |
|
- multi-label-classification |
|
- sentiment analysis |
|
- rubert |
|
- sentiment |
|
- bert |
|
- tiny |
|
- russian |
|
- multilabel |
|
- classification |
|
- prompt-classification |
|
--- |
|
|
|
Модель [RuBERT-tiny2](https://huggingface.co/cointegrated/rubert-tiny2) была fine-tuned для задачи __prompt classification__, предназначенная для __Russian__ текст. |
|
Выполняет задачу __multi-label classification__ со слудующимим категориями: |
|
|
|
```yaml |
|
0: write |
|
1: draw |
|
2: neutral |
|
``` |
|
|
|
Категории для русского языка: |
|
|
|
```yaml |
|
write: написать |
|
draw: рисовать |
|
neutral: нейтральность |
|
``` |
|
|
|
## Usage |
|
|
|
```python |
|
from transformers import pipeline |
|
model = pipeline(model="r1char9/rubert-tiny2-clf") |
|
model('Сгенерируй картину Томаса Шелби') |
|
# [{'label': 'draw', 'score': 0.8699279427528381}] |
|
``` |
|
|
|
## Metrics: |
|
|
|
``` |
|
metrics write draw neutral micro avg macro avg weighted avg |
|
precision 1.0 1.0 1.0 1.0 1.0 1.0 |
|
recall 1.0 1.0 1.0 1.0 1.0 1.0 |
|
f1-score 1.0 1.0 1.0 1.0 1.0 1.0 |
|
support 155.0 117.0 19.0 291.0 291.0 291.0 |
|
auc-roc 1.0 1.0 1.0 1.0 1.0 1.0 |
|
``` |
