predict_proba()

Возвращает вероятности классов.

Сигнатураclassifier.predict_proba(X)

Для классификаторов возвращает матрицу вероятностей принадлежности каждому классу; сумма по строке равна 1. Порядок столбцов соответствует classes_. Поддерживается не всеми моделями.

proba = clf.predict_proba(X_test)
print(proba[0])      # напр. [0.1, 0.9]
print(clf.classes_)
← Все записи: scikit-learn
Поддержать проект