add_effect_model: Add an additional model to the joint effect ensemble

In tidyhte: Tidy Estimation of Heterogeneous Treatment Effects

Add an additional model to the joint effect ensemble

Description

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

Usage

add_effect_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 joint effect 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_effect_model("SL.glm.interaction") -> hte_cfg

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