随机森林算法python代码
时间: 2023-08-29 18:12:14 浏览: 204
以下是使用Python实现随机森林算法的代码示例:
```python
from sklearn.ensemble import RandomForestClassifier
# 训练集和测试集
X_train, X_test, y_train, y_test = ...
# 创建随机森林分类器
rfc = RandomForestClassifier(n_estimators=100, max_depth=5)
# 训练模型
rfc.fit(X_train, y_train)
# 预测测试集
y_pred = rfc.predict(X_test)
# 评估模型
accuracy = rfc.score(X_test, y_test)
```
在上述代码中,我们使用`sklearn`库中的`RandomForestClassifier`类创建了一个随机森林分类器,并设置了100棵决策树和每棵树的最大深度为5。然后,我们使用训练集`X_train`和`y_train`来训练模型,使用测试集`X_test`进行预测并计算模型的准确度。
相关问题
随机森林回归Python代码
随机森林回归是一种机器学习算法,它可以利用Python实现。它使用一组决策树来生成预测结果,并通过平均法来提高准确性。 下面是一个简单的Python代码示例,用于实现随机森林回归:from sklearn.ensemble import RandomForestRegressor# 创建随机森林回归模型
regressor = RandomForestRegressor()# 训练模型
regressor.fit(X_train, y_train)# 预测结果
y_pred = regressor.predict(X_test)
阅读全文