python - 混淆矩阵预测标签

Question

我正在使用以下一段 python 代码来生成归一化的混淆矩阵

import scikitplot as skplt 
skplt.metrics.plot_confusion_matrix(y_test, y_pred, normalize=True)

这给了我这里显示的情节。但是，我想在 x 轴上旋转纹理（预测值，即中性、快乐、悲伤）。我希望旋转 x 轴上的预测值而不是水平线（如此处所示）并保持真实标签原样（水平）。我们如何在 Python 中做到这一点？

Answer 1

如果你想使用 scikitplot，我相信函数x_tick_rotation中有参数。因此，您可以将 xticks 旋转到 45 度，如下所示。plot_confusion_matrix

skplt.metrics.plot_confusion_matrix(y_test, y_pred, x_tick_rotation=45, normalize=True)