scikit-learn - 如何为sklearn热图混淆矩阵提供三种不同的配色方案（cmap值）

Question

我有一个 3*3 混淆矩阵（多标签分类），我想给每一行一个不同的颜色（以便可以根据颜色轻松区分每个类）如何使用 sklearn 热图功能做到这一点？

data=confusion_matrix(dt['Actual'], dt['Prediction'])
sns.heatmap(data,annot=True,cmap='BuGn',cbar= False)

标签： scikit-learndata-visualizationheatmapcolorbarconfusion-matrix