sklearn ：在随机森林分类器中获得预测和分数

Question

我能够获得 Random Forest 分类器中每一行的单独预测值。

有什么方法可以得到每一行的预测分数吗？

我能够得到整个数据的预测准确度分数。但是我每行都需要它。

Answer 1

只需比较数组即可轻松完成。无需逐行进行。

from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_classification

X, y = make_classification(n_samples=100, n_features=4,
                           n_informative=2, n_redundant=0,
                           random_state=0, shuffle=False)
clf = RandomForestClassifier(max_depth=2, random_state=0)
clf.fit(X, y)

predictions = clf.predict(X)
comparison = (pred==y)

print(comparison)
Output: 
array([ True,  True,  True,  True,  True,  True,  True,  True,  True,
   False,  True,  True,  True,  True,  True,  True, False,  True,
    True,  True, False,  True, False,  True, False,  True,  True,
    True,  True,  True,  True, False,  True,  True,  True,  True,
    True,  True,  True,  True,  True,  True,  True,  True,  True,
    True,  True,  True,  True,  True,  True,  True,  True,  True,
    True,  True,  True,  True,  True,  True,  True,  True,  True,
    True,  True,  True,  True,  True,  True,  True,  True,  True,
    True,  True,  True,  True,  True,  True,  True,  True,  True,
    True,  True, False,  True,  True,  True,  True,  True,  True,
    True, False,  True,  True,  True,  True,  True,  True,  True,
    True])

False 表示在这种情况下预测错误。

sklearn ：在随机森林分类器中获得预测和分数

sklearn : get prediction along with score in Random Forest classifier

machine-learning

random-forest

scikit-learn