predict_proba()
Возвращает вероятности классов.
Сигнатура
classifier.predict_proba(X)Для классификаторов возвращает матрицу вероятностей принадлежности каждому классу; сумма по строке равна 1. Порядок столбцов соответствует classes_. Поддерживается не всеми моделями.
proba = clf.predict_proba(X_test)
print(proba[0]) # напр. [0.1, 0.9]
print(clf.classes_)