Learn R Programming

causalDT (version 1.0.0)

predict_teacher_models: Predict wrappers for teacher models for causal distillation trees

Description

These functions are predict() method wrappers for various heterogeneous treatment effect learners that can be easily used as teacher models in the causal distillation tree framework.

  • predict_causal_forest(): wrapper around predict() for causal_forest() models.

  • predict_bcf(): wrapper around predict() for bcf() models.

Usage

predict_causal_forest(...)

predict_bcf(...)

Value

Vector of predicted conditional average treatment effects (CATEs).

Arguments

...

Additional arguments to pass to the base model predict functions.