hugo_train_model: Training binary classification model and choose the best one

In pbiecek/hugo: I'm your personal assistant!

Description

This function for the given formula and data builds logistic regressions models, decision trees, gbm model. It saves models to the models catalog and as a result returns trained models and evaluation of how these models are good.

Usage

hugo_train_model(data = NULL, formula = NULL)

Arguments

data	an optional data.frame containing the variables in the model
formula	formula describing the model to be fitted

Value

Function returns trained models and assessment of how these models are good.

Author(s)

Anna Kozak

Examples

## Not run: 
data("PimaIndiansDiabetes")
formula <- "diabetes~."
hugo_train_model(PimaIndiansDiabetes, formula)

## End(Not run)

pbiecek/hugo documentation built on May 12, 2019, 6:24 p.m.