python - 分别提取每个类的预测概率

Question

我想分别打印每个班级的概率，但我收到了这个错误：

TypeError: only integer scalar arrays can be converted to a scalar index 

这是我的代码：

classes = ('cat', 'dog', 'monkey', 'bird', 'wolf')

correct_pred = {classname: 0 for classname in classes}

total_pred = {classname: 0 for classname in classes}

for label, prediction in zip(y_test_argmax, y_pred_argmax):

    if label.any() == prediction.any():
        correct_pred[classes[label]] += 1
        total_pred[classes[label]] += 1


for classname, correct_count in correct_pred.items():

    accuracy = 100 * float(correct_count) / total_pred[classname]
    print("Accuracy for class {:5s} is: {:.1f} %".format(classname, accuracy))

标签： pythonmachine-learningmulticlass-classification