The model was trained on the VSMEC dataset, which contains 6,927 Vietnamese social media text samples annotated with emotion labels. The dataset includes the following emotion categories:

Enjoyment (0): Positive emotions, joy, happiness
Sadness (1): Sad, disappointed, gloomy feelings
Anger (2): Angry, frustrated, irritated
Fear (3): Scared, anxious, worried
Disgust (4): Disgusted, repelled
Surprise (5): Surprised, shocked, amazed
Other (6): Neutral or unclassified emotions

Results

The model was evaluated using the following metrics:

Accuracy: 0.0000
Macro-F1: 0.0000
Macro-Precision: 0.0000
Macro-Recall: 0.0000

Usage

You can use this model for emotion recognition in Vietnamese text. Below is an example of how to use it with the HuggingFace Transformers library:

python

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

# Load model and tokenizer
tokenizer = AutoTokenizer.from_pretrained("visolex/emotion-visobert")
model = AutoModelForSequenceClassification.from_pretrained("visolex/emotion-visobert")

# Example text
text = "Tôi rất vui vì hôm nay trời đẹp!"

# Tokenize
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)

# Predict
outputs = model(**inputs)
predicted_class = outputs.logits.argmax(dim=-1).item()

# Map to emotion name
emotion_map = {
    0: "Enjoyment",
    1: "Sadness",
    2: "Anger",
    3: "Fear",
    4: "Disgust",
    5: "Surprise",
    6: "Other"
}

predicted_emotion = emotion_map[predicted_class]
print(f"Text: {text}")
print(f"Predicted emotion: {predicted_emotion}")

License

This model is released under the Apache-2.0 license.

Acknowledgments

Base model: uitnlp/visobert
Dataset: VSMEC (Vietnamese Social Media Emotion Corpus)
ViSoLex Toolkit