Add an additional diagnostic to the effect model — add_effect

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".

Updated HTE_cfg object

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