pdpbox.pdp.PDPIsolate

class pdpbox.pdp.PDPIsolate(n_classes, which_class, feature, feature_type, feature_grids, percentile_info, display_columns, ice_lines, pdp, count_data, hist_data)

Save pdp_isolate results

Parameters:

n_classes: integer or None

number of classes for classifier, None when it is a regressor

which_class: integer or None

for multi-class classifier, indicate which class the result belongs to

feature: string or list

which feature is calculated on, list for one-hot encoding features

feature_type: string

type of the feature

feature_grids: list

feature grids

percentile_info: list

percentile information for feature grids

display_columns: list

columns to display as xticklabels

ice_lines: pandas DataFrame

ICE lines

pdp: 1-d numpy array

calculated PDP values

count_data: pandas DataFrame

data points distribution

hist_data: 1-d numpy array

data points distribution for numeric features