wonrax's picture
Update README.md
9076a58
metadata
language:
  - vi
tags:
  - sentiment
  - classification
license: mit
widget:
  - text: Không thể nào đẹp hơn
  - text: Quá phí tiền,  không đẹp
  - text: Cái này giá ổn không nhỉ?

GitHub Homepage

A model fine-tuned for sentiment analysis based on vinai/phobert-base.

Labels:

  • NEG: Negative
  • POS: Positive
  • NEU: Neutral

Dataset: 30K e-commerce reviews

Usage

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