Rubert Base Cased Sentiment Med

🚀 医療レビューの感情分析用RuBERT

このモデルは、医療レビューのコーパスで学習されたDeepPavlov/rubert-base-cased-conversationalモデルです。

🚀 クイックスタート

ラベル

0: 中立
1: 肯定的
2: 否定的

使用例

基本的な使用法

import torch
from transformers import AutoModelForSequenceClassification
from transformers import BertTokenizerFast

tokenizer = BertTokenizerFast.from_pretrained('blanchefort/rubert-base-cased-sentiment-med')
model = AutoModelForSequenceClassification.from_pretrained('blanchefort/rubert-base-cased-sentiment-med', 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

モデル学習に使用されたデータセット

Отзывы о медучреждениях

このデータセットには、医療機関に関するユーザーのレビューが含まれています。データセットは2019年5月にprodoctorov.ruサイトから収集されました。