Chipan/indonesia-5-emotion-cls-dataset
Viewer • Updated • 559 • 16
How to use Chipan/indobert-emotion with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Chipan/indobert-emotion") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Chipan/indobert-emotion")
model = AutoModelForSequenceClassification.from_pretrained("Chipan/indobert-emotion", device_map="auto")# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Chipan/indobert-emotion")
model = AutoModelForSequenceClassification.from_pretrained("Chipan/indobert-emotion", device_map="auto")This is a fine-tuned IndoBERT-based model for emotion classification in Bahasa Indonesia, designed to classify everyday informal or casual text into one of five emotional categories in bahasa Indonesia. The model is particularly useful for applications involving sentiment analysis, mental health monitoring, journaling platforms, or conversational AI.
The model is trained to classify input text into one of the following five emotions:
| Emotion | Description |
|---|---|
| Marah | Angry, Frustation |
| Sedih | Sadness, disappointment |
| Senang | Joy, excitement |
| Stress | Anxiety, mental pressure |
| Bersyukur | Gratitude, thankfulness |
Base model
indobenchmark/indobert-base-p1
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Chipan/indobert-emotion")