cleanDataLASSO: Clean LASSO data

In joeornstein/SRP: Stacked Regression and Poststratification

Description

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

Usage

cleanDataLASSO(trainset, lasso_vars, lasso_factors = NULL,
  testset = NULL, new_vars_lasso = NULL)

Arguments

trainset	The training dataset
lasso_vars	Predictor variables to use in the LASSO model
lasso_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_lasso	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.