我正在scikit-learn网站上跟踪此示例,以使用随机森林模型执行多输出分类。
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=5, n_features=5, n_informative=2, n_classes=2, random_state=1)
y2 = shuffle(y1, random_state=1)
Y = np.vstack((y1, y2)).T
forest = RandomForestClassifier(n_estimators=10, random_state=1)
multi_target_forest = MultiOutputClassifier(forest, n_jobs=-1)
multi_target_forest.fit(X, Y).predict(X)
print(multi_target_forest.predict_proba(X))
由此predict_proba
我得到2个5x2数组:
[array([[ 0.8, 0.2],
[ 0.4, 0.6],
[ 0.8, 0.2],
[ 0.9, 0.1],
[ 0.4, 0.6]]), array([[ 0.6, 0.4],
[ 0.1, 0.9],
[ 0.2, 0.8],
[ 0.9, 0.1],
[ 0.9, 0.1]])]
我真的很期待n_sample
通过n_classes
矩阵。我正在努力了解这与存在的类的概率之间的关系。
该文档的predict_proba
状态:
形状数组= [n_samples,n_classes],或者如果n_outputs> 1,则由n_outputs组成的数组列表。
输入样本的分类概率。类的顺序与属性classes_中的顺序相对应。
我想我的描述中有后者,但我仍在努力理解这与我的课堂概率有何关系。
此外,当我尝试访问模型的classes_
属性时,forest
我得到了,AttributeError
而该属性在上不存在MultiOutputClassifier
。如何将类与输出关联?
print(forest.classes_)
AttributeError: 'RandomForestClassifier' object has no attribute 'classes_'