R/recipe-api.R
add_propensity_score_model.Rd
This 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