dimkonn's picture
Update README.md
c00ec1d verified
|
Raw
History Blame Contribute Delete
706 Bytes
metadata
library_name: transformers
tags:
  - russian
  - toxicity
  - text-classification
  - roberta
license: mit
language:
  - ru
datasets:
  - Mnwa/russian-toxic
base_model:
  - ai-forever/ru-en-RoSBERTa
pipeline_tag: text-classification

Russian Toxic Classifier

Модель бинарной классификации токсичности текстов на базе ai-forever/ru-en-RoSBERTa.

Метрики (test)

  • Accuracy: 0.9186
  • F1: 0.9162
  • ROC AUC: 0.9765

Использование

from transformers import pipeline
classifier = pipeline("text-classification", model="dimkonn/russian-toxic-classifier")
result = classifier("Вы что, совсем тупой?")
print(result)