无法在Python中打印完整的混淆矩阵

Question

我正在建立图像预测模型。

为了更好地了解精度，我正在使用scikit learn打印混淆矩阵。

当我使用30个类别时，混淆矩阵完全显示在打印屏幕上，但是当我将模型用于50个类别时，混淆矩阵就开始打印截断/缩小的混淆矩阵版本。

我尝试了多种方式将其打印完整或写入文件中，但是没有用。

输出是这样的：

[[0 0 0 ... 0 0 0]
[0 0 0 ... 0 0 0]
[0 0 0 ... 0 0 0]
...
[0 0 0 ... 0 0 0]
[0 0 0 ... 0 0 0]
[0 0 0 ... 0 0 0]]

我也尝试过np.set_printoptions。但它也不起作用。

使用np.set_printoptions甚至不能让我打印任何东西，并且会出现类似typeError can not implement between int and str的错误。

Answer 1

您可以通过设置打印完整的数组而不会被截断

np.set_printoptions(threshold=np.inf)

或

np.set_printoptions(threshold=np.nan)

默认情况下，大于1000个元素的数组将被截断。

有关完整文档，请参见http://docs.scipy.org/doc/numpy/reference/generated/numpy.set_printoptions.html。

Answer 2

您可以将数组转换为列表，然后打印：

my_list = my_array.tolist()