cleanDataGBM: Clean GBM Data

In joeornstein/SRP: Stacked Regression and Poststratification

Description

Cleans the training data for xgboost, which requires a model matrix instead of a formula

Usage

cleanDataGBM(trainset, gbm_vars, gbm_factors = NULL, testset = NULL,
  new_vars_gbm = NULL)

Arguments

trainset	The training dataset
gbm_vars	Predictor variables to use in the GBM model
gbm_factors	Which predictor variables need to be converted to factors
testset	Optional test set; included so that both cleaned versions will have the same dimensions
new_vars_gbm	Optional variable names to include as input

Value

A list, containing cleaned versions of the trainset and testset

joeornstein/SRP documentation built on Oct. 15, 2020, 8:30 p.m.