Lrnr_cate_DR Class

This class defines a doubly-robust (DR) meta-learner of the conditional average treatment effect.

Format

An R6 class with public methods to initialize the learner, create a regression task, and access the base learner.

Details

In the supported binary/categorical-treatment setting, this learner targets the conditional mean difference over the chosen modifier set V, namely E[Y(1) - Y(0) | V].

Super classes

sl3::Lrnr_base -> hte3::Lrnr_hte -> Lrnr_cate_DR

Methods

Public methods

• Lrnr_cate_DR$new()

• Lrnr_cate_DR$get_pseudo_data()

• Lrnr_cate_DR$clone()

Inherited methods

• sl3::Lrnr_base$assert_trained()
• sl3::Lrnr_base$base_chain()
• sl3::Lrnr_base$base_predict()
• sl3::Lrnr_base$base_train()
• sl3::Lrnr_base$chain()
• sl3::Lrnr_base$custom_chain()
• sl3::Lrnr_base$get_outcome_range()
• sl3::Lrnr_base$get_outcome_type()
• sl3::Lrnr_base$predict()
• sl3::Lrnr_base$predict_fold()
• sl3::Lrnr_base$print()
• sl3::Lrnr_base$process_formula()
• sl3::Lrnr_base$reparameterize()
• sl3::Lrnr_base$retrain()
• sl3::Lrnr_base$sample()
• sl3::Lrnr_base$set_train()
• sl3::Lrnr_base$subset_covariates()
• sl3::Lrnr_base$train()
• sl3::Lrnr_base$train_sublearners()
• hte3::Lrnr_hte$check_treatment_type()
• hte3::Lrnr_hte$get_modifiers()
• hte3::Lrnr_hte$make_metalearner_task()

---

Method new()

Usage

Lrnr_cate_DR$new(
  base_learner,
  treatment_level = NULL,
  control_level = NULL,
  ...
)

Arguments

base_learner

A sl3 learner object inheriting from Lrnr_base that specifies the base supervised learning algorithm used by the meta-learner.

treatment_level

A treatment level encoding the treatment assignment of interest.

control_level

A treatment level encoding the control (or reference) treatment assignment.

---

Method get_pseudo_data()

Usage

Lrnr_cate_DR$get_pseudo_data(
  hte3_task,
  treatment_level = NULL,
  control_level = NULL,
  ...
)

Arguments

treatment_level

A treatment level encoding the treatment assignment of interest.

control_level

A treatment level encoding the control (or reference) treatment assignment.

---

Method clone()

The objects of this class are cloneable with this method.

Usage

Lrnr_cate_DR$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.