随机森林回归python代码
时间: 2023-05-20 22:02:52 浏览: 102
随机森林的python代码
以下是一个简单的随机森林回归的Python代码示例:
```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`是随机种子,`X_train`和`y_train`是训练数据集,`X_test`是测试数据集,`y_pred`是预测结果。
阅读全文