TrainModel: TrainModel

Description Usage Arguments Value

View source: R/TrainModel.R

Description

Makes a sample enriched for cases using ideas from surrogate guided sampling strategy

Usage

1
TrainModel(train.df, Y = "Y", feat = "X", surr = "Z", model = "glm")

Arguments

train.df

Data frame containing training data

Y

Column name of the binary outcome

model

Whether to use a glm, lasso, or ridge model

Value

A list of coef: estimated betas n_cases: Number of cases in the training sample auc_train: Estimated training AUC train.model: The trained model object


wlktan/surrogateSampling documentation built on May 23, 2019, 3:07 p.m.