sklearn.inspection.PartialDependenceDisplay?

部分依賴圖(PDP)可視化。

建議使用 plot_partial_dependence創建一個 PartialDependenceDisplay。所有參數都存儲為屬性。

部分依賴的高級繪圖用戶指南中閱讀更多內容 。

0.22版中的新功能。

參數 說明
pd_results list of (ndarray, ndarray)
featurespartial_dependence結果。每個元組對應一個(平均預測值,網格)。
features list of (int,) or list of (int, int)
給定圖的特征指標。一個整數的元組將繪制一個特征的部分依賴曲線。兩個整數的元組將繪制一條雙向偏相關曲線作為輪廓圖。
feature_names list of str
features中的索引相對應的特征名稱。
target_idx int
- 在多分類設置中,指定計算PDP的類別。請注意,對于二分類,始終使用正類(索引為1)。
- 在多輸出設置中,指定應為其計算PDP的任務。

在二進制分類或經典回歸設置中被忽略。
pdp_lim dict
全局最小和最大平均預測,這樣所有圖將具有相同的比例和y限制。pdp_lim[1]是單個部分相關曲線的全局最小值和最大值。pdp_lim[2]是雙向偏相關曲線的全局最小值和最大值。
deciles dict
features中的十進制的特征索引。
屬性 說明
bounding_ax_ matplotlib Axes or None
如果ax是軸或無,則bounding_ax_是繪制偏相關圖網格的軸。如果ax是軸列表或軸的numpy數組,bounding_ax_則為None。
axes_ ndarray of matplotlib Axes
如果ax是軸或無,則axes_[i, j]是第i行和第j列上的軸。如果ax是軸列表,則axes_[i]ax中的第i個項目。無元素對應于該位置上不存在的軸。
lines_ ndarray of matplotlib Artists
如果ax是軸或無,則lines_[i, j]是第i行和第j列的偏相關曲線。如果ax是軸列表, 則lines_[i]是與ax中的第i個項目相對應的偏相關曲線。無元素對應于不存在的軸或不包含折線圖的軸。
deciles_vlines_ ndarray of matplotlib LineCollection
如果ax是軸或無,則vlines_[i, j]是表示第i行第j列的x軸極坐標的線集合。如果ax是軸列表,則vlines_[i]對應于ax中的第i個項目 。無元素對應于不存在的軸或不包含PDP圖的軸。0.23版本添加.
deciles_hlines_ ndarray of matplotlib LineCollection
如果ax是軸或無,則vlines_[i, j]是表示第i行第j列的y軸極坐標的線集合。如果ax是軸列表,則vlines_[i]對應于ax中的第i個項目 。無元素對應于不存在的軸或不包含2向圖的軸。0.23版本添加.
contours_ ndarray of matplotlib Artists
如果ax是軸或無,則contours_[i, j]是在第i行和第j列上的偏相關圖。如果ax是軸列表, 則contours_[i]是與ax中的第i個項目相對應的偏相關圖。無元素對應于不存在的軸或不包含輪廓圖的軸。
figure_ matplotlib Figure
該圖包含部分依賴圖。

方法

方法 說明
plot([ax, n_cols, line_kw, contour_kw]) 繪制部分依賴圖。
__init__(pd_results, *, features, feature_names, target_idx, pdp_lim, deciles)

[源碼]

初始化self,參見help(type(self))獲取更準確的說明。

plot(ax=None, n_cols=3, line_kw=None, contour_kw=None)

繪制部分依賴圖。

參數 說明
ax Matplotlib axes or array-like of Matplotlib axes, default=None
- 如果傳入單個軸,則將其視為邊界軸,并在這些邊界內繪制部分依賴圖網格。n_cols參數控制網格中的列數。
- 如果傳入一個類數組的軸,則部分依賴圖將直接繪制到這些軸中。
- 如果None,則創建圖形和邊界軸,并將其視為單軸情況。
n_cols int, default=3
網格圖中的最大列數。僅在 ax為單軸或None時有效。
line_kw dict, default=None
對于單向偏相關圖,包含傳遞給matplotlib.pyplot.plot的關鍵字字典。
contour_kw dict, default=None
對于雙向部分依賴圖,包含傳遞給matplotlib.pyplot.contourf的關鍵字字典。
返回值 說明
display PartialDependenceDisplay

sklearn.inspection.PartialDependenceDisplay使用示例?