python随机森林评估方法
时间: 2023-12-14 17:34:06 浏览: 114
针对Python中随机森林的评估方法,可以采用以下步骤:
1.使用sklearn.ensemble中的RandomForestClassifier类来构建随机森林分类器模型。
2.使用模型的fit()方法来拟合训练数据。
3.使用模型的predict()方法来对测试数据进行预测。
4.使用模型的score()方法来计算模型在测试数据上的准确率。
5.使用模型的oob_score_属性来计算模型的Out-of-Bag误差。
下面是一个简单的例子,演示如何使用随机森林分类器进行评估:
```python
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
# 生成随机数据集
X, y = make_classification(n_samples=1000, n_features=4, random_state=0)
# 将数据集分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)
# 构建随机森林分类器模型
clf = RandomForestClassifier(n_estimators=100, max_depth=2, random_state=0)
# 拟合训练数据
clf.fit(X_train, y_train)
# 对测试数据进行预测
y_pred = clf.predict(X_test)
# 计算模型在测试数据上的准确率
accuracy = clf.score(X_test, y_test)
print("Accuracy:", accuracy)
# 计算模型的Out-of-Bag误差
oob_error = 1 - clf.oob_score_
print("OOB error:", oob_error)
```
阅读全文