treatment_balance_curve(uplift, treatment, winsize)¶
Compute the treatment balance curve: proportion of treatment group in the ordered predictions.
- uplift (1d array-like) – Predicted uplift, as returned by a model.
- treatment (1d array-like) – Treatment labels.
- winsize (int) – Size of the sliding window for calculating the balance between treatment and control.
Points on a curve.
array (shape = [>2]), array (shape = [>2])