train_it: Generate Prediction Model

In CausalGPS: Matching on Generalized Propensity Scores with Continuous Exposures

Description

Function to develop prediction model based on user's preferences.

Usage

train_it(target, input, pred_model, sl_lib_internal = NULL, ...)

Arguments

target	A vector of target data.
input	A vector, matrix, or dataframe of input data.
pred_model	Prediction model algorithm. 'sl': SuperLearner The required parameters: sl_lib: a set of methods used for estimating target value (e.g., ("SL.xgboost","SL.earth","SL.gam","SL.ranger"))
sl_lib_internal	The internal library to be used by SuperLearner
...	Model related parameters should be provided.

Value

prediction model

CausalGPS documentation built on Feb. 7, 2022, 5:12 p.m.