Я работаю над проблемой моделирования склонности рекламной кампании. Мой набор данных состоит из пользователей, которые исторически нажимали на объявления и тех, кто не нажимал.
Чтобы измерить производительность моей модели, я построил кумулятивные усиления и графики подъема, используя sklearn. Ниже приведен код для того же самого:
import matplotlib.pyplot as plt
import scikitplot as skplt
Y_test_pred_ = model.predict_proba(X_test_df)[:]
skplt.metrics.plot_cumulative_gain(Y_test, Y_test_pred_)
plt.show()
skplt.metrics.plot_lift_curve(Y_test, Y_test_pred_)
plt.show()
На графике, который я получаю, показаны графики как для пользователей класса 0, так и для пользователей класса 1 
Мне нужно построить только кривую класса 1 относительно базовой линии. Есть ли способ, которым я могу это сделать?