Phobert Base Vietnamese Sentiment

🚀 wonrax/phobert-base-vietnamese-sentiment

このモデルは、vinai/phobert-base をベースに、感情分析用にファインチューニングされたものです。越南語の文章の感情を分類することができます。

🚀 クイックスタート

GitHubホームページ

このモデルは、vinai/phobert-base をベースに感情分析用にファインチューニングされています。

ラベル:

• NEG: ネガティブ
• POS: ポジティブ
• NEU: 中立

データセット: 30Kの電子商取引レビュー

💻 使用例

基本的な使用法

import torch
from transformers import RobertaForSequenceClassification, AutoTokenizer

model = RobertaForSequenceClassification.from_pretrained("wonrax/phobert-base-vietnamese-sentiment")

tokenizer = AutoTokenizer.from_pretrained("wonrax/phobert-base-vietnamese-sentiment", use_fast=False)

# Just like PhoBERT: INPUT TEXT MUST BE ALREADY WORD-SEGMENTED!
sentence = 'Đây là mô_hình rất hay , phù_hợp với điều_kiện và như cầu của nhiều người .'  

input_ids = torch.tensor([tokenizer.encode(sentence)])

with torch.no_grad():
    out = model(input_ids)
    print(out.logits.softmax(dim=-1).tolist())
    # Output:
    # [[0.002, 0.988, 0.01]]
    #     ^      ^      ^
    #    NEG    POS    NEU

📄 ライセンス

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