This adds a diagnostic to the effect model.

add_effect_diagnostic(hte_cfg, diag)

Arguments

hte_cfg

HTE_cfg object to update.

diag

Character indicating the name of the diagnostic to include. Possible values are "MSE", "RROC" and, for SuperLearner ensembles, "SL_risk" and "SL_coefs".

Value

Updated HTE_cfg object

Examples

library("dplyr")
basic_config() %>%
   add_effect_diagnostic("RROC") -> hte_cfg