class BaseTLearner[source]

BaseTLearner(learner=None, control_learner=None, treatment_learner=None, ate_alpha=0.05, control_name=0) :: BaseLearner

A parent class for T-learner regressor classes.

A T-learner estimates treatment effects with two machine learning models.

Details of T-learner are available at Kunzel et al. (2018) (https://arxiv.org/abs/1706.03461).

class BaseTRegressor[source]

BaseTRegressor(learner=None, control_learner=None, treatment_learner=None, ate_alpha=0.05, control_name=0) :: BaseTLearner

A parent class for T-learner regressor classes.

class BaseTClassifier[source]

BaseTClassifier(learner=None, control_learner=None, treatment_learner=None, ate_alpha=0.05, control_name=0) :: BaseTLearner

A parent class for T-learner classifier classes.

class XGBTRegressor[source]

XGBTRegressor(ate_alpha=0.05, control_name=0, *args, **kwargs) :: BaseTRegressor

A parent class for T-learner regressor classes.

class MLPTRegressor[source]

MLPTRegressor(ate_alpha=0.05, control_name=0, *args, **kwargs) :: BaseTRegressor

A parent class for T-learner regressor classes.