model_train {OptHoldoutSize}

Train model (wrapper)

Description

Train model using either a GLM or a random forest

Usage

model_train(train_data, model_family = "log_reg", ...)

Arguments

train_data

Data to use for training; assumed to have one binary column called Y

model_family

Either 'log_reg' for logistic regression or 'rand_forest' for random forest

...

Passed to function glm() or ranger()

Value

Fitted model of type GLM or Ranger

Examples

# See examples for model_predict