Russian Toxicity Classifier

Модель для бинарной классификации русскоязычных текстов.

Задача

Определение токсичности текста.

Классы

0 — нетоксичный текст
1 — токсичный текст

Базовая модель

ai-forever/ru-en-RoSBERTa

Метрики

Accuracy: 0.9995 Precision: 0.9989 Recall: 1.0000 F1: 0.9995 MCC: 0.9989 ROC_AUC: 1.0000

Пример использования

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="molyalya/ru-toxicity-classifier"
)

result = classifier("Вы что, совсем тупой?")
print(result)