Sets the function responsible for plotting the predictions generated from a fitted model.

set_plot_predictions(algorithm, family)

Arguments

algorithm

A character vector of length one; the algorithm to run on the data. Choices are one of c("glmnet", "random_forest", "support_vector_machine").

family

A character vector of length one; the type of regression to run on the data. Choices are one of c("gaussian", "binomial"). Defaults to "gaussian".

Value

A function; the function for plotting the predictions generated from a fitted model.

See also

Other setters: set_categorical_variables, set_column_names, set_cores, set_dependent_variable, set_independent_variables, set_looper_, set_looper, set_measure, set_parallel, set_plot_model_performance, set_preprocess, set_random_state, set_resample