Knn算法进行房价预测代码
时间: 2023-07-26 10:40:17 浏览: 80
以下是基于Python的KNN算法进行房价预测的代码示例:
```python
import numpy as np
from sklearn.neighbors import KNeighborsRegressor
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error
# 准备数据集
data = np.genfromtxt('house_price.csv', delimiter=',', skip_header=1)
X = data[:, :-1]
y = data[:, -1]
# 分割训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 训练模型
k = 5 # 选择K值
model = KNeighborsRegressor(n_neighbors=k)
model.fit(X_train, y_train)
# 预测并评估模型
y_pred = model.predict(X_test)
mse = mean_squared_error(y_test, y_pred)
print('均方误差:', mse)
```
在这个示例中,我们使用numpy库读取CSV格式的房价数据集,并将其分为训练集和测试集。然后,我们选择K=5来训练KNN模型,并使用均方误差评估模型的准确性。最后,我们输出模型的均方误差。
阅读全文