如何从函数中打印 sklearn confusion_matrix 输出?

How to print sklearn confusion_matrix output from within a function?

confusion_matrix 在我笔记本上的命令行中正常工作,但我似乎无法在函数内部编写时打印输出。在这两种情况下,它都应用于相同的数组。

我是不是漏掉了什么?

如果您在笔记本单元格中调用一个函数作为最后一个命令,输出将自动打印,但是当它被包装在一个函数中时情况并非如此。在这种情况下,您必须在函数内的某处调用 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)