波士顿房价预测python代码_波士顿房价预测——线性模型（numpy实施）,实现

数据准备

下载数据集：https://archive.ics.uci.edu/ml/machine-learning-databases/housing/housing.data

代码实现参考百度 AI Studio实现

github地址:(内包含数据集)

https://github.com/Classmate-Huang/Boston-Housing-Forecast

代码1

梯度下降

import numpy as np

import matplotlib.pyplot as plt

def load_data():

# 导入房价数据

datafile = 'housing.data'

data = np.fromfile(datafile, sep=' ')

# 将原始数据Reshape 并且拆分成训练集和测试集

data = data.reshape([-1, 14])

offset = int(data.shape[0]*0.8)

train_data = data[:offset]

# 归一化处理

maximums, minimums, avgs = train_data.max(axis=0), train_data.min(axis=0), train_data.sum(axis=0) / train_data.shape[0]

for i in range(14):

data[:, i] = (data[:, i] - avgs[i]) / (maximums[i] - minimums[i])

train_data = data[:offset]