Форма передаваемых значений (2, 49), индексы подразумевают (2, 137210)

Question

Я пытаюсь построить кривую подъема, используя определенную функцию ниже:

  def plotLiftChart(actual, predicted):
df_dict = {'actual': X, 'pred': Y}
df = pd.DataFrame(df_dict)
pred_ranks = pd.qcut(df['pred'].rank(method='first'), 100, labels=False)
actual_ranks = pd.qcut(df['actual'].rank(method='first'), 100, labels=False)
pred_percentiles = df.groupby(pred_ranks).mean()
actual_percentiles = df.groupby(actual_ranks).mean()
plt.title('Lift Chart')
plt.plot(np.arange(.01, 1.01, .01), np.array(pred_percentiles['pred']),
         color='darkorange', lw=2, label='Prediction')
plt.plot(np.arange(.01, 1.01, .01), np.array(pred_percentiles['actual']),
         color='navy', lw=2, linestyle='--', label='Actual')
plt.ylabel('Target Percentile')
plt.xlabel('Population Percentile')
plt.xlim([0.0, 1.0])
plt.ylim([-0.05, 1.05])
plt.legend(loc="best")

Затем я пытаюсь вызвать функцию для целевых значений обучающего набора: Y_test и прогнозируемых вероятностей: Pred,Оба имеют одинаковую форму: (137210,)

plotLiftChart(Y_test, pred)

Когда я пытаюсь сделать это, я получаю эту ошибку: форма переданных значений (2, 49), индексы подразумевают (2, 137210).Кто-нибудь знает, как это решить, пожалуйста?