绘制混淆矩阵

Question

我正在尝试按照以下代码段使用scikitplot.metrics.plot_confusion_matrix绘制混淆矩阵-

import scikitplot as skplt

Y_Test = [1, 0, 1, 0, 1, 1, 0, 1, 1 ,1 ,1, 1, 0, 0 ,0];
Y_Pred = [1, 0, 0, 0, 1, 1, 0, 1, 1 ,1 ,1, 1, 1, 0 ,0];
cm = skplt.metrics.plot_confusion_matrix(Y_Test,Y_Pred,normalize=True, text_fontsize = 'large')

但是，我并没有获得如图所示的预期结果（某种程度上，文本与轴重叠）。如何使它整洁干净？

Answer 1

首先检查安装的matplotlib的版本。问题出在scikitplot内部使用的“ imshow”功能。在修复之前，请先卸载当前的matplotlib版本，然后使用以下命令安装“ 3.0.3”版本。

pip uninstall matplotlib
pip install matplotlib=='3.0.3'