Views
No views yet
cointegrated/rubert-tiny2.ЭкономикаСпортИз жизниИнтернет и СМИКультураДомБывший СССР69-я параллельМирНаука и техникаПутешествияРоссияНацпроектыСиловые структурыЦенности| Метрика | Значение |
|---|---|
| Training Loss | 1.752300 |
| Validation Loss | 1.319999 |
| Accuracy | 0.679245 |
| F1-score | 0.616440 |
1from transformers import pipeline
2
3classifier = pipeline(
4 "text-classification",
5 model="cointegrated/rubert-tiny2",
6 tokenizer="cointegrated/rubert-tiny2"
7)
8
9# Классификация текста
10text = "Ваш текст для классификации здесь"
11result = classifier(text)
12
13label_map = {
14 'LABEL_0': 'Экономика',
15 'LABEL_1': 'Спорт',
16 'LABEL_2': 'Из жизни',
17 'LABEL_3': 'Интернет и СМИ',
18 'LABEL_4': 'Культура',
19 'LABEL_5': 'Дом',
20 'LABEL_6': 'Бывший СССР',
21 'LABEL_7': '69-я параллель',
22 'LABEL_8': 'Мир',
23 'LABEL_9': 'Наука и техника',
24 'LABEL_10': 'Путешествия',
25 'LABEL_11': 'Россия',
26 'LABEL_12': 'Нацпроекты',
27 'LABEL_13': 'Силовые структуры',
28 'LABEL_14': 'Ценности',
29}
30
31print(f"Категория: {label_map[raw_output[0]['label']]}, Уверенность: {raw_output[0]['score']:.4f}")