model_train: Train model (wrapper)

View source: R/OptHoldoutSize_simulation.R

model_trainR Documentation

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

OptHoldoutSize documentation built on April 12, 2025, 1:34 a.m.