线性回归
:用来确定两种或两种以上变量间的相互依赖的定量关系
房价与房屋尺寸的线性拟合
:对房屋成交信息建立回归方程,并依据回归方程对房屋价格进行预测
数据格式一览
源代码
import matplotlib.pyplot as plt
import numpy as np
from sklearn.linear_model import LinearRegression
//读取数据集
datasets_X = []
datasets_Y = []
fr = open('prices.txt','r')
lines = fr.readlines()
for line in lines:
items = line.strip().split(',')
//存储房屋尺寸
datasets_X.append(int(items[0]))
//存储房屋价格
datasets_Y.append(int(items[1]))
length = len(datasets_X)
//将其变为二维数组,以符合线性回归拟合函数输入参数的要求
datasets_X = np.array(datasets_X).reshape(length,1)
//将其变为等长数组
datasets_Y = np.array(datasets_Y)
minX = min(datasets_X)
maxX = max(datasets_X)
//建立等差数列,方便后续画图
X = np