线性回归
1 一元线性回归 import pandas as pd import numpy as np import matplotlib.pyplot as plt from sklearn.linear_model import LinearRegression #导入机器学习库中的线性回归模块 data=pd.DataFrame({'square_feet':[150,200,250,300,350,400,600], 'price':[5450,6850,8750,9650,10450,13450,16450]}) #创建一组7行2列的数据,square_feet为房屋面积,price为对应价格 data_train=np.array(data['square_feet']).reshape(data['square_feet'].shape[0],1)#这里是将数据转化为一个1维矩阵 print(type(data_train)) print(data_train) data_test=data['price'] #创建线性回归模型,拟合面积与价格并通过面积预测价格 regr=LinearRegression() #创建线性回归模型,参数默认 # 这里的fit()中的第一个参数必须是二维数组才行,否则报错 regr.fit(data_train,data_test)#拟合数据