如何从函数内打印sklearn confusion_matrix输出？

Question

confusion_matrix在笔记本电脑上的命令行上可以正常工作，但是当写入函数中时，似乎无法使其输出其输出。在两种情况下都将其应用于相同的阵列。

我想念什么吗？

Answer 1

如果将函数作为笔记本单元中的最后一个命令调用，则输出将自动打印，但是将其包装在函数中时并非如此。在这种情况下，您必须在函数内的某个位置调用print（）函数。

import numpy as np
from sklearn.metrics import confusion_matrix

a = np.array([1,0,1,1,0,1])
b = np.array([0,1,0,1,0,0])

def show_confusion(a, b):
    print(confusion_matrix(a, b))

show_confusion(a, b)