Rubert Conversational Cased Sentiment

🚀 Keras模型結合ruBERT對話嵌入器用於情感分析

本項目主要用於對俄語文本進行情感分類。模型基於 Tatyana/ru_sentiment_dataset 數據集進行訓練。

🚀 快速開始

安裝依賴

!pip install tensorflow-gpu
!pip install deeppavlov
!python -m deeppavlov install squad_bert
!pip install fasttext
!pip install transformers
!python -m deeppavlov install bert_sentence_embedder

加載模型並進行預測

from deeppavlov import build_model

model = build_model(Tatyana/rubert_conversational_cased_sentiment/custom_config.json)
model(["Сегодня хорошая погода", "Я счастлив проводить с тобою время", "Мне нравится эта музыкальная композиция"])

✨ 主要特性

• 支持對俄語文本進行情感分類。
• 基於ruBERT對話嵌入器，能夠更好地處理對話場景下的文本。

📚 詳細文檔

標籤含義