python实现波士顿房价预测
时间: 2023-10-08 15:08:06 浏览: 104
好的,我可以为你介绍一下如何使用 Python 实现波士顿房价预测。
首先,你需要导入需要的库,包括 pandas、numpy、matplotlib 和 sklearn:
```python
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error
```
接着,我们可以使用 `load_boston()` 函数来加载波士顿房价数据集,并将其存储为 Pandas DataFrame 对象:
```python
boston = load_boston()
boston_df = pd.DataFrame(boston.data, columns=boston.feature_names)
boston_df['MEDV'] = boston.target
```
现在我们可以进行数据预处理,包括数据的标准化和分割为训练集和测试集:
```python
X = boston_df.iloc[:, :-1].values
y = boston_df.iloc[:, -1].values
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=0)
# 数据标准化
from sklearn.preprocessing import StandardScaler
sc = StandardScaler()
X_train = sc.fit_transform(X_train)
X_test = sc.transform(X_test)
```
接下来,我们可以使用线性回归模型来训练数据,并使用均方根误差来评估模型的准确性:
```python
lin_reg = LinearRegression()
lin_reg.fit(X_train, y_train)
y_pred = lin_reg.predict(X_test)
mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
print('均方根误差:', round(rmse, 2))
```
最后,我们可以使用 `matplotlib` 库来绘制预测值和实际值之间的散点图:
```python
plt.scatter(y_test, y_pred)
plt.xlabel('实际值')
plt.ylabel('预测值')
plt.show()
```
这样,我们就完成了波士顿房价预测的 Python 实现。希望能对你有所帮助!
阅读全文