我在一个.py文件中创建了一个模型，并保存了它的.pkl文件，以便之后进行分析。这个模型是使用来自Kaggle情感数据集的代码构建的https://www.kaggle.com/shivamburnwal/speech-emotion-recognition。问题是，当我使用这个代码的模型来检测新音频的情感时，输出是以独热编码格式显示的。有没有办法可以得到实际的情感（如’快乐’、’恐惧’等）而不是1和0？

回答：

预测输出是一个类似这样的列表：

prediction = [0,0,0,0,1,0,0] , [0,1,0,0,0,0,0], [0,0,0,1,0,0,0]

然后你可以使用以下代码将它们转换：

pred = []for x in prediction:    if x == [0,0,0,0,1,0,0]:        pred.append('Sad')    elif x == [0,1,0,0,0,0,0]:        pred.append('Happy')    elif x == [0,0,0,1,0,0,0]:        pred.append('disgust')print(pred)

输出为：

['Sad', 'Happy', 'disgust']

请根据需要添加更多的elif语句。