predict_CDI: Predict Probability of CDI

In PippintheFoolhardy/SimFunctions: CDI Simulator R Functions

Description

Use machine-learning methods to predict probability of agent having CDI

Usage

predict_CDI(
  df,
  cores = 1L,
  folds = 5L,
  model = "gbm",
  scoring = "ROC",
  repeats = 2L,
  trees = 1L,
  grid = 0L
)

Arguments

df	Table of agent data to use in predictive model
cores	Number of cores to use in parallel
folds	Number of cross-validation folds
model	Machine-learning model to use
scoring	Method of scoring to judge model accuracy
repeats	Number of repeats for cross-validation
trees	Number of trees to use if applicable to machine-learning model
grid	Tuning grid to use if applicable

Value

Machine-learning model

Examples

model <- predict_CDI(df, cores = 1L, folds = 5L, model = "gbm", scoring = "ROC", repeats = 2L, trees = 1L, grid = 0L);

PippintheFoolhardy/SimFunctions documentation built on Dec. 18, 2021, 7:43 a.m.