Downloads · 30 days
5
15% of all-time downloads
grafnatt/ru-en-toxicity-classifier-rosberta
ru-en-toxicity-classifier-rosberta is a text classification model from grafnatt. Use it when you need a label for a piece of text. It is set up for transformers.
Модель предназначена для классификации токсичных, грубых и агрессивных сообщений на русском и английском языках. Обучена на базе предобученной модели ai-forever/ru-en-RoSBERTa с использованием библиотеки PyTorch Light…
Downloads · 30 days
5
15% of all-time downloads
All-time downloads
34
Public
Parameters
405M
3.2 GB on disk
Likes
0
Public
Click a slice to open those files.
.ckpt1.6 GB · 50%
From the Hugging Face model README
Модель предназначена для классификации токсичных, грубых и агрессивных сообщений на русском и английском языках. Обучена на базе предобученной модели ai-forever/ru-en-RoSBERTa с использованием библиотеки PyTorch Lightning.
Модель показала выдающиеся результаты на тестовой выборке:
from transformers import pipeline
classifier = pipeline("text-classification", model="grafnatt/ru-en-toxicity-classifier-rosberta")
result = classifier("Вы что, совсем тупой? Я же уже объяснял!")
print(result)
# [{'label': 'LABEL_1', 'score': 0.7924}]