add_propensity_score_model: Add an additional model to the propensity score ensemble

In tidyhte: Tidy Estimation of Heterogeneous Treatment Effects

Add an additional model to the propensity score ensemble

Description

This adds a learner to the ensemble used for estimating propensity scores.

Usage

add_propensity_score_model(hte_cfg, model_name, ...)

Arguments

hte_cfg	HTE_cfg object to update.
model_name	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.

Value

Updated HTE_cfg object

Examples

library("dplyr")
basic_config() %>%
   add_propensity_score_model("SL.glmnet", alpha = c(0, 0.5, 1)) -> hte_cfg

tidyhte documentation built on Aug. 14, 2023, 5:08 p.m.