Learn R Programming

OptHoldoutSize (version 0.1.0.1)

model_train: Train model (wrapper)

Description

Train model using either a GLM or a random forest

Usage

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

Value

Fitted model of type GLM or Ranger

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()

Examples

Run this code

# See examples for model_predict

Run the code above in your browser using DataLab