Python在混淆矩阵图上调整尺度强度

Question

我有一些混淆矩阵图，它们的数字总和不相同(这些数字在基准测试中为100分)请参阅下面附带的示例图像:混淆矩阵我不希望22和32有相同的颜色强度，但在相同的规模从0到100的强度水平。我如何在python中调整比例给出以下使用的代码:

import matplotlib.pyplot as plt
from sklearn.datasets import make_classification
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
cm = confusion_matrix(y_true, y_pred, labels=["Up", "Down"])
disp = ConfusionMatrixDisplay(confusion_matrix=cm,display_labels=["Up", "Down"])
disp.plot(cmap="OrRd")

Answer 1

disp.ax_.get_images()[0].set_clim(0, 100)

和完整的代码:

import matplotlib.pyplot as plt
from sklearn.datasets import make_classification
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
cm = confusion_matrix(y_true, y_pred, labels=["Up", "Down"])
disp = ConfusionMatrixDisplay(confusion_matrix=cm,display_labels=["Up", "Down"])
disp.plot(cmap="OrRd")
disp.ax_.get_images()[0].set_clim(0, 100)