应用python写出随机森林波士顿房价回归预测
时间: 2023-07-08 16:23:39 浏览: 142
基于线性回归实现波士顿房价预测的python源码大作业.zip
5星 · 资源好评率100%
好的,以下是基于Python语言的随机森林波士顿房价回归预测示例代码:
```
# 导入所需的库
import numpy as np
import pandas as pd
from sklearn.ensemble import RandomForestRegressor
from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
# 加载数据集
boston = load_boston()
X = pd.DataFrame(boston.data, columns=boston.feature_names)
y = pd.Series(boston.target)
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
# 构建随机森林回归模型
rf = RandomForestRegressor(n_estimators=100, random_state=42)
rf.fit(X_train, y_train)
# 预测测试集结果
y_pred = rf.predict(X_test)
# 计算模型的平均绝对误差(MAE)和均方误差(MSE)
mae = np.mean(abs(y_pred - y_test))
mse = np.mean((y_pred - y_test) ** 2)
# 输出模型的评估结果
print("MAE:", round(mae, 2))
print("MSE:", round(mse, 2))
```
在这个例子中,我们使用了scikit-learn库中的随机森林回归器(RandomForestRegressor)实现波士顿房价回归预测。我们首先从sklearn.datasets中加载数据集,然后将其分成训练集和测试集。接下来,我们使用RandomForestRegressor构建随机森林回归模型,并使用fit方法拟合训练数据。最后,我们使用predict方法对测试数据进行预测,并计算模型的MAE和MSE评估指标。
阅读全文