Russian Toxicity Classifier

🚀 ロシア語毒性コメント分類器

BERTベースの分類器で、ロシア語の毒性コメントを分類することができます。Conversational Rubertをファインチューニングし、複数のデータセットで学習されています。

🚀 クイックスタート

このモデルは、BERTベースの分類器で、ロシア語の毒性コメントを分類するために学習されています。Conversational Rubertをベースモデルとして使用し、複数のデータセットを統合して学習されています。

モデル情報

評価指標

テストデータセットで得られた評価指標は以下の通りです。

💻 使用例

基本的な使用法

from transformers import BertTokenizer, BertForSequenceClassification

# load tokenizer and model weights
tokenizer = BertTokenizer.from_pretrained('s-nlp/russian_toxicity_classifier')
model = BertForSequenceClassification.from_pretrained('s-nlp/russian_toxicity_classifier')

# prepare the input
batch = tokenizer.encode('ты супер', return_tensors='pt')

# inference
model(batch)

📚 ドキュメント

引用

このモデルを使用する場合は、以下の引用を使用してください。

@article{dementieva2022russe,
  title={RUSSE-2022: Findings of the First Russian Detoxification Shared Task Based on Parallel Corpora},
  author={Dementieva, Daryna and Logacheva, Varvara and Nikishina, Irina and Fenogenova, Alena and Dale, David and Krotova, Irina and Semenov, Nikita and Shavrina, Tatiana and Panchenko, Alexander}
}

📄 ライセンス

このモデルはOpenRAIL++ライセンスの下で提供されています。このライセンスは、公共の利益に役立つ産業や学術分野の様々な技術開発をサポートしています。