Я хочу вычислить и напечатать roc_auc_score, чтобы оценить производительность моей модели (случайный лес), я выполняю НЛП, поэтому мои данные в y_test и y_pred в основном представляют собой список слов, я векторизую их функция pipe_vect.transform, но когда я запускаю тренировку, у меня появляется следующая ошибка:
ValueError: Only one class present in y_true. ROC AUC score is not defined in that case."
Я думаю, что ошибка появляется, потому что y_test и y_pred не имеют одинаковое измерение.
Мой код:
x_test_vect = pipe_vect.transform(x_test)
y_pred = model.predict_proba(x_test_vect)
auc_score = roc_auc_score(y_test, y_pred)
print('Performance model :', auc_score)