README.md

---
language:
- vi
base_model:
- VietAI/vit5-base
---
A model fine-tuned for sentiment analysis based on [VietAI/vit5-base](https://huggingface.co/vinai/phobert-base).

Labels:
- NEG: Negative
- POS: Positive
- NEU: Neutral

Dataset: Comments on Shoppe (https://shopee.vn/)
## Usage
```python
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("lamsytan/sentiment-analysis-base-phobert")
model = AutoModelForSequenceClassification.from_pretrained("lamsytan/sentiment-analysis-base-phobert")

sentence = "Áo đẹp lắm nhá lần sau sẽ ghé tiếp ạ"

inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True)

with torch.no_grad():
    outputs = model(**inputs)
    logits = outputs.logits
    probabilities = torch.softmax(logits, dim=-1)

print(probabilities.tolist())
    # Output:
    # [[0.010827462188899517, 0.9538241624832153, 0.035348404198884964]]
    #         ^                         ^                  ^