class BaseSLearner[source]

BaseSLearner(learner=None, ate_alpha=0.05, control_name=0) :: BaseLearner

A parent class for S-learner classes. An S-learner estimates treatment effects with one machine learning model. Details of S-learner are available at Kunzel et al. (2018) (https://arxiv.org/abs/1706.03461).

class BaseSRegressor[source]

BaseSRegressor(learner=None, ate_alpha=0.05, control_name=0) :: BaseSLearner

A parent class for S-learner regressor classes.

class BaseSClassifier[source]

BaseSClassifier(learner=None, ate_alpha=0.05, control_name=0) :: BaseSLearner

A parent class for S-learner classifier classes.