[转CS-自学] MELD复现，两类class_weight为0，怎么解决

转码

🔗

无辜起尘埃 2021-7-20 01:20:08 | 只看该作者 |倒序浏览

本楼：

👍 0

👎

全局：

6 0

注册一亩三分地论坛，查看更多干货！

您需要登录才可以下载或查看附件。没有帐号？注册账号

x

抱歉打扰，CS/EE小白，复现代码时候有问题（https://github.com/declare-lab/MELD/issues/10），开贴很多人复现出现这类问题，请问怎么解决，希望能给点建议，谢谢
Hi, I have tried the bc_LSTM baseline with bimodal in emotion classification, but the F1-score and accuracy of 'fear' and 'disgust' are always zero, so I can't reproduce the result in paper.https://github.com/declare-lab/MELD/issues/10

The command I use:
python baseline.py -classify emotion -modality bimodal -train
The results:
      precision recall  f1-score support
   0    0.7322 0.7795 0.7551    1256
   1    0.4799 0.4662 0.4729    281
   2    0.0000 0.0000 0.0000       50
   3    0.2781 0.2019 0.2340    208
   4    0.4813 0.5448 0.5111    402
   5    0.0000 0.0000 0.0000       68
   6    0.3832 0.4377 0.4087    345
The emotion labels:
Emotion - {'neutral': 0, 'surprise': 1, 'fear': 2, 'sadness': 3, 'joy': 4, 'disgust': 5, 'anger': 6}.