Skip to contents

This function helps to train the ML model across ensemble members in parallel.

Usage

parallel_train(
  full_data,
  method = "randomForest",
  cores = parallel::detectCores()
)

Arguments

full_data

numeric: the matrix generated using the `prepare_train_dat` function.

method

character: machine learning method (currently support randomForest and xgboost).

cores

numeric: how many CPus to be used in the calculation, the default is the total CPU number you have.

Value

list of trained models across ensemble members.

Author

Dongchen Zhang