CoolFace
Modelpublic

alter-wang/luke-japanese-large-wrime-emotion-alpha

sourceHugging Faceupdated 1y agoView on Hugging Face
1likes29downloads
Model Card

This is a LUKE (Japanese version) Base model for emotion analysis in Japanese additionally fine-tuned for emotion detection and classification.

The model was based on Mizuiro-sakura/luke-japanese-large-sentiment-analysis-wrime, and later fine-tuned on a dataset containing 10 labels of emotional texts from Japanese dictionaries, blogs and Twitter.

The dataset was composed of about 1,700 sentences, with about 100~250 sentences each for each emotion category.

emotion_mapping = { 0: 'amaze/surprise', 1: 'anger', 2: 'dislike', 3: 'excitement', 4: 'fear', 5: 'joy', 6: 'like/fondness', 7: 'relief', 8: 'sad/gloom', 9: 'shame' }

emotion_mapping = { 0: '驚き', 1: '怒り', 2: 'いや', 3: '昂り', 4: '怖がり', 5: '喜び', 6: '好き', 7: '安らぎ', 8: '悲しみ', 9: '恥ずかしい' }

Evaluation result with our test data

AccuracyMacro F1
95.07%94.97%

I will continue to fine-tune this model soon.