pdpbox.pdp.PDPInteract

class pdpbox.pdp.PDPInteract(n_classes, which_class, features, feature_types, feature_grids, pdp_isolate_outs, pdp)

Save pdp_interact 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

features: list

[feature1, feature2]

feature_types: list

[feature1 type, feature2 type]

feature_grids: list

[feature1 grid points, feature2 grid points]

pdp_isolate_outs: list

[feature1 pdp_isolate result, feature2 pdp_isolate result]

pdp: pandas DataFrame

calculated PDP values for each gird combination