python 随机森林模型
时间: 2024-01-07 20:23:17 浏览: 106
python随机森林优化模型.py
5星 · 资源好评率100%
随机森林是一种集成学习方法,它由多个决策树组成。每个决策树都是独立训练的,然后通过投票或平均的方式来进行预测。随机森林在分类和回归问题中都有很好的表现。
以下是一个使用随机森林模型的例子:
```python
from sklearn.ensemble import RandomForestRegressor
from sklearn import metrics
import numpy as np
# 创建随机森林回归模型
forest = RandomForestRegressor(n_estimators=100, random_state=1, n_jobs=-1)
# 使用训练数据进行模型训练
forest.fit(X_train, Y_train)
# 使用验证数据进行预测
y_validation_pred = forest.predict(X_validation)
# 评估模型性能
score = forest.score(X_validation, Y_validation)
print('随机森林模型得分:', score)
print('Mean Absolute Error:', metrics.mean_absolute_error(Y_validation, y_validation_pred))
print('Mean Squared Error:', metrics.mean_squared_error(Y_validation, y_validation_pred))
print('Root Mean Squared Error:', np.sqrt(metrics.mean_squared_error(Y_validation, y_validation_pred)))
```
这段代码中,我们首先导入了`RandomForestRegressor`类和一些评估指标的函数。然后,我们创建了一个随机森林回归模型,并使用训练数据进行训练。接下来,我们使用验证数据进行预测,并计算了模型的准确率分值、MAE、MSE和RMSE等评估指标。
阅读全文