Rubert Base Cased Russian Sentiment

🚀 RuBERTモデル - 短いロシア語テキストの感情分類

このモデルは、短いロシア語テキストの 感情分類 用にファインチューニングされた RuBERT モデルです。このタスクは、以下のラベルを持つ 多クラス分類 です。

0: neutral
1: positive
2: negative

ラベルのロシア語表記は以下の通りです。

neutral: нейтральный
positive: позитивный
negative: негативный

🚀 クイックスタート

💻 使用例

基本的な使用法

from transformers import pipeline
model = pipeline(model="seara/rubert-base-cased-russian-sentiment")
model("Привет, ты мне нравишься!")
# [{'label': 'positive', 'score': 0.9818321466445923}]

📚 ドキュメント

📦 データセット

このモデルは、以下のデータセットの統合データで学習されました。

• Kaggle Russian News Dataset
• Linis Crowd 2015
• Linis Crowd 2016
• RuReviews
• RuSentiment

学習データの概要は、S. SmetaninのGithubリポジトリ で確認できます。

Smetaninによって収集されたすべてのロシア語感情データセットのダウンロードリンクは、この リポジトリ で確認できます。

🔧 学習詳細

学習は、この プロジェクト で以下のパラメータを使用して行われました。

tokenizer.max_length: 256
batch_size: 32
optimizer: adam
lr: 0.00001
weight_decay: 0
epochs: 2

学習/検証/テストデータの分割比率は、80%/10%/10% です。

📊 評価結果 (テストデータセット)

📄 ライセンス

このモデルは、MITライセンスの下で提供されています。