为什么颜色条未标准化（0 到 1）？怎么强求？

Question

我正在绘制一个混淆矩阵。我使用了 ScikitLearn 的功能。但是不知道为什么colorbar没有0到1的范围，有没有办法强制呢？

import itertools
def plot_confusion_matrix(cm, title='Confusion matrix RF', cmap=plt.cm.viridis):
plt.imshow(cm, interpolation='nearest', cmap=cmap) 
plt.title(title)
plt.colorbar() 
tick_marks = np.arange(len(np.unique(y))) 

plt.xticks(tick_marks, rotation=90)
ax = plt.gca()
ax.set_xticklabels(['s'+lab for lab in (ax.get_xticks()+1).astype(str)])
plt.yticks(tick_marks)
ax.set_yticklabels(['s'+lab for lab in (ax.get_yticks()+1).astype(str)])

plt.tight_layout()
plt.ylabel('True label')
plt.xlabel('Predicted label')




cm_imp = confusion_matrix(y_true, y_pred)
cm_imp_normalized = cm_imp.astype('float') / cm_imp.sum(axis=1)[:, np.newaxis] 
plt.figure(figsize=(8,6)) 
plot_confusion_matrix(cm_imp_normalized)
plt.show() 
print("")
print("")

Answer 1

您可以使用vmin, vmax 参数将颜色范围设置为imshow。

plt.imshow(data, cmap=cmap, vmin=0, vmax=1)