Update README.md
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ tags:
|
|
| 21 |
пайплайне контроля качества — анализ исходящих ответов операторов в реальном времени
|
| 22 |
перед отправкой клиенту.
|
| 23 |
|
| 24 |
-
- **Вход:** текст сообщения на русском языке
|
| 25 |
- **Выход:** класс (0 — нетоксично, 1 — токсично) и вероятность токсичности
|
| 26 |
- **Базовая модель:** ai-forever/ru-en-RoSBERTa (энкодер заморожен, обучалась только классификационная голова)
|
| 27 |
-
|
|
@@ -39,7 +39,7 @@ roc_auc: 0.99
|
|
| 39 |
|
| 40 |
## Пример использования
|
| 41 |
|
| 42 |
-
|
| 43 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 44 |
import torch
|
| 45 |
|
|
@@ -61,4 +61,4 @@ toxicity_prob = probs[0][1].item()
|
|
| 61 |
|
| 62 |
print(f"Класс: {prediction}")
|
| 63 |
print(f"Вероятность токсичности: {toxicity_prob:.4f}")
|
| 64 |
-
|
|
|
|
| 21 |
пайплайне контроля качества — анализ исходящих ответов операторов в реальном времени
|
| 22 |
перед отправкой клиенту.
|
| 23 |
|
| 24 |
+
- **Вход:** текст сообщения на русском языке
|
| 25 |
- **Выход:** класс (0 — нетоксично, 1 — токсично) и вероятность токсичности
|
| 26 |
- **Базовая модель:** ai-forever/ru-en-RoSBERTa (энкодер заморожен, обучалась только классификационная голова)
|
| 27 |
-
|
|
|
|
| 39 |
|
| 40 |
## Пример использования
|
| 41 |
|
| 42 |
+
```python
|
| 43 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 44 |
import torch
|
| 45 |
|
|
|
|
| 61 |
|
| 62 |
print(f"Класс: {prediction}")
|
| 63 |
print(f"Вероятность токсичности: {toxicity_prob:.4f}")
|
| 64 |
+
```
|