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

log*_*ick 2 python machine-learning scikit-learn

我正在关注本教程: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))
Run Code Online (Sandbox Code Playgroud)

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

Fla*_*ino 6

您可以获得训练集预测:

y_train_pred = clf.predict(X_train) 
Run Code Online (Sandbox Code Playgroud)

clf你的安装位置在哪里RandomForestClassifier。之后您可以使用

metrics.accuracy_score(y_train, y_train_pred) 
Run Code Online (Sandbox Code Playgroud)

来计算训练精度。或者,您可以使用

clf.score(X_train, y_train)
Run Code Online (Sandbox Code Playgroud)

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