wayfair / pylift

Uplift modeling package.
http://pylift.readthedocs.io
BSD 2-Clause "Simplified" License
368 stars 76 forks source link

Calculate actual and predicted treatment effect #42

Closed mwijaya3 closed 4 years ago

mwijaya3 commented 4 years ago

Thanks for the great package. I got a model on a set of features, a conversion column (0, 1), and a treatment indicator (0, 1). I am interested in plotting a decile of predicted vs actual treatment effect for hold-out set.

I am struggling to understand how I can calculate the predicted and the actual treatment effect based on the model. Any pointers?

From my understanding, the actual average treatment effect (ATE) can be calculated by the difference in mean outcomes of treatment - control. How can I get predicted average treatment effect? My hunch is that I can take the average of the uplift score per decile. Is that correct?

Thanks in advance!