Bangkah/atha-text-dataset
Viewer • Updated • 1.8k • 55 • 1
How to use Bangkah/atha-text-classifier with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Bangkah/atha-text-classifier") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Bangkah/atha-text-classifier")
model = AutoModelForSequenceClassification.from_pretrained("Bangkah/atha-text-classifier")# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Bangkah/atha-text-classifier")
model = AutoModelForSequenceClassification.from_pretrained("Bangkah/atha-text-classifier")Model ini adalah fine-tuned indobenchmark/indobert-base-p1 untuk klasifikasi sentimen Bahasa Indonesia 3 kelas.
Label output:
negativeneutralpositiveTraining data: https://huggingface.co/datasets/Bangkah/atha-text-dataset
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model_id = "Bangkah/atha-text-classifier"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)
text = "produk ini bagus dan pengirimannya cepat"
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)
with torch.no_grad():
logits = model(**inputs).logits
probs = torch.softmax(logits, dim=-1)[0]
label_id = int(torch.argmax(probs).item())
label = model.config.id2label[label_id]
score = float(probs[label_id].item())
print({"label": label, "confidence": round(score, 4)})
| true\pred | negative | neutral | positive |
|---|---|---|---|
| negative | 100 | 0 | 0 |
| neutral | 0 | 100 | 0 |
| positive | 0 | 0 | 100 |
precision recall f1-score support
negative 1.0000 1.0000 1.0000 100
neutral 1.0000 1.0000 1.0000 100
positive 1.0000 1.0000 1.0000 100
accuracy 1.0000 300
macro avg 1.0000 1.0000 1.0000 300
weighted avg 1.0000 1.0000 1.0000 300
Base model
indobenchmark/indobert-base-p1
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Bangkah/atha-text-classifier")