Sets the function responsible for measuring the results.

set_measure(measure = NULL, algorithm, family)

Arguments

measure

A function; the function for measuring the results. Defaults to NULL.

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 measuring the results.

See also

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