fx_model: Train/Test Model

View source: R/fx_model.R

fx_modelR Documentation

Train/Test Model

Description

Train/test model on sub data frames

Usage

fx_model(
  df.set,
  covar = NULL,
  voi = NULL,
  outcome = NULL,
  model.type = "logistic",
  z.pred = F
)

Arguments

df.set

list containing train and test data frames

outcome

df0 column name for outcome measure to be predicted (string)

model.type

machine learning model ('rf', 'logistic', 'regression', 'rf.regression', 'svm') (string)

z.pred

standardize predictive features (boolean)

Value

A list of length three, containing the following elements:

  • "pred.covar": data frame of predicted values from covariate model

  • "pred.full": data frame of predicted values from full model

  • "parameters": model parameters


fishpm/nruPredict documentation built on July 12, 2022, 3:22 p.m.