sklearn随机森林
时间: 2023-09-05 15:07:02 浏览: 98
随机森林是一种集成学习方法,利用多个决策树进行集成,通常用于分类和回归问题。在sklearn库中,可以使用`sklearn.ensemble`模块中的`RandomForestClassifier`和`RandomForestRegressor`来构建随机森林模型。
对于分类问题,可以使用`RandomForestClassifier`类来构建随机森林分类器。以下是一个示例代码:
```python
from sklearn.ensemble import RandomForestClassifier
# 创建一个随机森林分类器对象
rf = RandomForestClassifier(n_estimators=100, random_state=42)
# 训练模型
rf.fit(X_train, y_train)
# 使用模型进行预测
y_pred = rf.predict(X_test)
```
对于回归问题,可以使用`RandomForestRegressor`类来构建随机森林回归器。以下是一个示例代码:
```python
from sklearn.ensemble import RandomForestRegressor
# 创建一个随机森林回归器对象
rf = RandomForestRegressor(n_estimators=100, random_state=42)
# 训练模型
rf.fit(X_train, y_train)
# 使用模型进行预测
y_pred = rf.predict(X_test)
```
在上述示例代码中,`n_estimators`参数表示随机森林中决策树的数量,可以根据需要进行调整。`random_state`参数用于设置随机种子,以保证结果的可复现性。
阅读全文