sklearn.inspection.PartialDependenceDisplay?
部分依賴圖(PDP)可視化。
建議使用 plot_partial_dependence
創建一個 PartialDependenceDisplay
。所有參數都存儲為屬性。
0.22版中的新功能。
參數 | 說明 |
---|---|
pd_results | list of (ndarray, ndarray)features 的partial_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 | dictfeatures 中的十進制的特征索引。 |
屬性 | 說明 |
---|---|
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 |