Out of memory issue when call uplift_auc_score about scikit-uplift HOT 3 CLOSED

Hello! 🖐️

Thanks for the feedback, could you tell me which version of scikit-upllift you are using?

You can find out the version using the following command:

import sklift

print(sklift.__version__)

from scikit-uplift.

Hello, thanks for response. here are some libs version and python 3.7.5
numpy vesrion: 1.19.5
pandas version: 1.1.5
scikit uplift vesrion: 0.5.1

I did some debug with the source code, looks the perfect uplift will return a matrix instead of an 1d array. is it possible some data can make below code return a n*n matrix?

   y_true, treatment = np.array(y_true), np.array(treatment)

    cr_num = np.sum((y_true == 1) & (treatment == 0))  # Control Responders
    tn_num = np.sum((y_true == 0) & (treatment == 1))  # Treated Non-Responders

    # express an ideal uplift curve through y_true and treatment
    summand = y_true if cr_num > tn_num else treatment
    
    perfect_uplift = 2 * (y_true == treatment) + summand

from scikit-uplift.

the issue was solved.

from scikit-uplift.