| 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.