set_plot_predictions: Set plot predictions function.

In easyml: Easily Build and Evaluate Machine Learning Models

Description

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

Usage

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

easyml documentation built on June 26, 2017, 9:02 a.m.