Chipan/indonesia-5-emotion-cls-dataset
Viewer β’ Updated β’ 559 β’ 25
π This model is a fine-tuned version of indobenchmark/indobert-base-p2 on an Indonesian emotion classification dataset.
It is trained to classify user-written daily mood/feeling texts into several emotional categories.
indobenchmark/indobert-base-p2The model is trained to predict the following emotions:
You can use this model directly with the transformers library:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
tokenizer = AutoTokenizer.from_pretrained("Atherizz/emolog-indobert")
model = AutoModelForSequenceClassification.from_pretrained("Atherizz/emolog-indobert")
inputs = tokenizer("aku capek dan bingung hari ini", return_tensors="pt")
outputs = model(**inputs)
predicted_class = torch.argmax(outputs.logits)
print(predicted_class)