Python 調整混淆矩陣的尺度強度 plot

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)