got predict_proba of shape (431, 1), but need classifier with two classes for roc_auc_score scoring

Question

I am trying to plot a learning curve for my prediction model that is Random Classifier, where my X looks like the following:

    0            1          2