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