Downloads · 30 days
11
44% of all-time downloads
siridech24/wangchanberta-sentiment-fine-tuned
wangchanberta-sentiment-fine-tuned is a machine learning model from siridech24. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This is a finetuned sentiment classifier based on WangchanBERTa, trained on Thai customer messages from the TISCO Insure platform.
Downloads · 30 days
11
44% of all-time downloads
All-time downloads
25
Public
Parameters
105M
421 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors421 MB · 99%
From the Hugging Face model README
This is a finetuned sentiment classifier based on WangchanBERTa, trained on Thai customer messages from the TISCO Insure platform.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
tokenizer = AutoTokenizer.from_pretrained("siridech24/sentiment")
model = AutoModelForSequenceClassification.from_pretrained("siridech24/sentiment")
inputs = tokenizer("ประทับใจบริการมากครับ", return_tensors="pt")
outputs = model(**inputs)
pred = torch.argmax(outputs.logits, dim=1)
print(pred.item())