maks-sh / scikit-uplift

:exclamation: uplift modeling in scikit-learn style in python :snake:
https://www.uplift-modeling.com
MIT License
725 stars 96 forks source link

Incorrect error message in metrics.uplift_at_k #209

Open dennism-tulcolabs opened 1 year ago

dennism-tulcolabs commented 1 year ago

In line 468, the message for the ValueError raised when n_trmnt > n_samples_trmnt says ...f'in the treatment group: {n_samples_ctrl}'. Instead of n_samples_ctrl, the message should reference n_samples_trmnt.