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.
- Return type
array (shape = [>2]), array (shape = [>2])