#python #scikit-learn #classification #multilabel-classification
#python #scikit-learn #классификация #multilabel-классификация
Вопрос:
Я хочу получить вероятности прогнозирования для каждого класса каждого вывода. Но classes_
атрибут не существует в MultiOutputClassifier.
Как я могу связать классы с выводом?
from sklearn.datasets import make_classification
from sklearn.multioutput import MultiOutputClassifier
from sklearn.ensemble import RandomForestClassifier
from sklearn.utils import shuffle
import numpy as np
X, y1 = make_classification(n_samples=16, n_features=8, n_informative=4, n_classes=4, random_state=1)
y2 = shuffle(y1, random_state=1)
Y = np.vstack((y1, y2)).T
forest = RandomForestClassifier(n_estimators=16, random_state=1)
multi_target_forest = MultiOutputClassifier(forest, n_jobs=-1)
multi_target_forest.fit(X, Y).predict(X)
multi_target_forest.predict_proba(X)
multi_target_forest.classes_
Ошибка атрибута: объект ‘MultiOutputClassifier’ не имеет атрибута ‘classes_’
Ответ №1:
Используйте estimators_
атрибут MultiOutputClassifier
для доступа к атрибутам оценщика.
Попробуйте это!
['y_{} class_{}'.format(idx,klass) for idx,forest in enumerate(multi_target_forest.estimators_)
for klass in forest.classes_]
#output:
['y_0 class_0',
'y_0 class_1',
'y_0 class_2',
'y_0 class_3',
'y_1 class_0',
'y_1 class_1',
'y_1 class_2',
'y_1 class_3']