predict_ICE.Plotter {distillML}R Documentation

Prediction Function for ICE Plots

Description

Gives predictions at each point on the grid.

Usage

predict_ICE.Plotter(object, features = NULL, save = TRUE)

Arguments

object

The Interpeter object to use.

features

A vector with the names of the features to predict ICE plots for

save

A boolean indicator to indicate whether the calculations should be saved in the interpreter object or not. This can help reduce computation if the ICE functions are used many times, but requires additional memory to store the predictions. By default, this is TRUE.

Value

A list of data frames, one for each feature. In each data frame, the first column contains the grid values for the feature, and each subsequent column has a single observation corresponding to the prediction of the model when with the given feature set to that grid point value.

Note

This method is meant to primarily be used to find the exact values for the ICE curves plotted. Note that after the PDP curve is plotted, the returned object of this function will be the saved predictions for plotting the curve, rather than a recalculation of the values.


[Package distillML version 0.1.0.13 Index]