R/recipe-api.R
add_propensity_score_model.RdThis adds a learner to the ensemble used for estimating propensity scores.
add_propensity_score_model(hte_cfg, model_name, ...)HTE_cfg object to update.
Character indicating the name of the model to
incorporate into the propensity score ensemble. Possible values
use SuperLearner naming conventions. A full list is available
with SuperLearner::listWrappers("SL")
Parameters over which to grid-search for this model class.
Updated HTE_cfg object
library("dplyr")
basic_config() %>%
add_propensity_score_model("SL.glmnet", alpha = c(0, 0.5, 1)) -> hte_cfg