python - 如何获得Scikit-learn RandomForest的训练精度？

Question

我正在关注本教程:https://www.datacamp.com/community/tutorials/random-forests-classifier-python关于将 Scikit-learn 与随机森林结合使用。但是，当前代码仅显示测试准确性，而我还想知道训练准确性，因为可能的数据集非常小。

获取测试准确率的代码为:

from sklearn import metrics
print("Accuracy:", metrics.accuracy_score(y_test, y_pred))

我如何修改它以获得训练准确性？

Answer 1

您可以获得训练集预测:

y_train_pred = clf.predict(X_train) 

其中clf是您安装的RandomForestClassifier。之后就可以使用

metrics.accuracy_score(y_train, y_train_pred) 

计算训练准确率。或者，您可以使用

clf.score(X_train, y_train)

这应该会给你相同的结果。