我能够在随机森林分类器中获取每行的单个预测值。
有没有什么方法可以获取每行的预测分数呢?
我能够获取整个数据的预测准确率分数。但是我需要每行的分数单独获取。
回答:
可以通过简单地比较数组来实现这一点,不需要逐行操作。
from sklearn.ensemble import RandomForestClassifierfrom sklearn.datasets import make_classificationX, 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表示在该情况下预测是错误的。