Rubert Base Cased Sentiment Rurewiews

🚀 用於產品評論情感分析的RuBERT模型

本項目是基於 DeepPavlov/rubert-base-cased-conversational 模型，在 RuReviews 數據集上進行訓練得到的，可用於對產品評論進行情感分析。

✨ 主要特性

• 支持對產品評論進行情感分類，包含中性、積極、消極三種情感標籤。

📦 安裝指南

文檔未提供安裝步驟，跳過此章節。

💻 使用示例

基礎用法

import torch
from transformers import AutoModelForSequenceClassification
from transformers import BertTokenizerFast

tokenizer = BertTokenizerFast.from_pretrained('blanchefort/rubert-base-cased-sentiment-rurewiews')
model = AutoModelForSequenceClassification.from_pretrained('blanchefort/rubert-base-cased-sentiment-rurewiews', return_dict=True)

@torch.no_grad()
def predict(text):
    inputs = tokenizer(text, max_length=512, padding=True, truncation=True, return_tensors='pt')
    outputs = model(**inputs)
    predicted = torch.nn.functional.softmax(outputs.logits, dim=1)
    predicted = torch.argmax(predicted, dim=1).numpy()
    return predicted

📚 詳細文檔

標籤說明

模型訓練數據集

RuReviews

RuReviews：一個自動標註的俄語產品評論情感分析數據集。

🔧 技術細節

文檔未提供技術實現細節，跳過此章節。

📄 許可證

文檔未提供許可證信息，跳過此章節。