This adds a learner to the ensemble used for estimating a model of the conditional expectation of the outcome.

add_outcome_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 outcome 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_outcome_model("SL.glm.interaction") -> hte_cfg