Вывод предиката_проба в scikit-learn

Question

Предположим, у меня есть выборка данных с двумя классами, помеченными 0 и 1. Когда я запускаю output = clf.predict_proba(X_input), каждая строка в output состоит из 2 столбцов, соответствующих вероятности каждого класса.

Представляет ли первый столбец вероятность класса 0 или 1? Метод Foret_proba GradientBoostingClassier говорит:

"Вероятности классов входных выборок. Порядок классов соответствует тому, что указано в атрибуте classes_."

Означает ли это, что 0 или 1 является первым элементом выборки данных соответствует первому столбцу в выводе predict_proba?

Answer 1

Как правило, классификатор будет иметь атрибут с именем classes_, который будет заполняться после подгонки и хранения классов. Порядок вывода метода predict_proba будет таким же, как порядок в этом атрибуте.

Например:

nb = MultinomialNM()
nb.fit(some_gender_data)
nb.classes_
array(['F', 'M'], dtype='<U1')

Насколько мне известно, у всех классификаторов в sklearn этот атрибут когда-то подходит.