机器学习实验——单变量线性回归（披萨价格预测问题）

实验内容

假设某披萨店的披萨价格和披萨直径之间有下列数据关系：

根据上面的训练数据，预测12英寸的披萨的可能售价。
1、直径为自变量X，价格为因变量y，画出二者的散点图，并给出结论。
2、根据现有的训练数据求线性回归模型，并画出拟合直线，给出拟合直线方程。
3、预测12英寸披萨的价格。
4、评价模型的准确率，分析模型预测结果
注：

测试数据：

实验内容

一、首先进行绘制散点图，绘制散点图我们使用matplotlib.pyplot库，直径、价格分别为自变量x，y，并且设置所画散点图的属性。

代码：

import matplotlib.pyplot as pltdata_x = [[6], [8], [10], [14], [18]]data_y = [[7], [9], [13], [17.5], [18]]plt.title("Pizza Price vs Diameter")# 设置标题名称plt.xlim(0, 25)# 设置x轴坐标范围plt.ylim(0, 25)# 设置y轴坐标范围plt.xlabel('Diameter') # 设置x轴标题plt.ylabel('Price')# 设置y轴标题plt.plot(data_x, data_y, 'k.') #绘制散点图plt.show()

运行结果：

注：

plt.plot(data_x, data_y, 'k.')

中'k.'为图形符号以及颜色风格的设置，现给出常用部分速查表：

二、使用sklearn.linear_model.LinearRegression对象进行拟合，首先创建一个线性回归对象，使用fit方法进行训练模型，再使用intercept_``lin_reg.coef_获得所得拟合方程的截距与斜率。为了方便绘制拟合方程图像，创建一个二位列表(data_x2)方便将方程直线绘制范围为增加到[0,25]，再使用predict函数进行预测出所对应的y值，最后使用plot函数绘制拟合方程。

代码：

data_x2 = [[0], [10], [15], [25]]model = linear_model.LinearRegression()model.fit(data_x, data_y)intercept = str(model.intercept_[0])[:4]coef = str(model.coef_[0][0])[:4]print("截距为：" + intercept)print("斜率为：" + coef)print("拟合直线方程为：" + "y=" + coef + "x+" + intercept)data_y2 = model.predict(data_x2)plt.plot(data_x2, data_y2, '-')plt.show()

运行结果：

看完代码的小伙伴们可能会有疑问为什么会出现这样的代码

intercept = str(model.intercept_[0])[:4]coef = str(model.coef_[0][0])[:4]

其实不难理解，我们先输出一下'model.intercept_' 'model.coef_'的输出结果：

是的，你没看错，'model.intercept_'输出的数据是一个列表，'model.coef_'更是一个二位列表，而题干要求我们输出拟合方程，如果截距和斜率居然用列表的形式表示出来，那岂不是成了四不像！所以我先将他们转化成字符串，最后保留小数点后两位数字。

三、在第二步我们已经通过model对象训练出模型，预测12英寸披萨的价格那就是再简单不过的事情了，我们只需要使用predict方法直接预测！

代码：

piece = str(model.predict([[12]])[0][0])[:5]print("预测12英寸披萨的价格为:" + piece)

运行结果：

别再问我'str(model.predict([[12]])[0][0])[:5]'这是什么意思，往上翻！！！！

四、第四步就是考察你之前是否明白之前的所讲内容，话不多说，首先传入测试数据，画出散点图，使用之前的训练模型进行数据预测，最后使用model.score方法对模型准确率进行计算。

总代码：

import matplotlib.pyplot as pltfrom sklearn import linear_modeldef run_plt():plt.rcParams['font.sans-serif'] = ['SimHei'] # 标题中文不报错plt.title("Pizza Price vs Diameter")plt.xlim(0, 25)plt.ylim(0, 25)plt.xlabel('Diameter')plt.ylabel('Price')return pltif __name__ == "__main__":# 题目一plt = run_plt()data_x = [[6], [8], [10], [14], [18]]data_y = [[7], [9], [13], [17.5], [18]]plt.plot(data_x, data_y, 'k.')plt.show()# 题目二plt = run_plt()plt.plot(data_x, data_y, 'k.')data_x2 = [[0], [10], [15], [25]]model = linear_model.LinearRegression()model.fit(data_x, data_y)print("截距为：" + str(model.intercept_[0])[:4])print("斜率为：" + str(model.coef_[0][0])[:4])data_y2 = model.predict(data_x2)plt.plot(data_x2, data_y2, '-')plt.show()# 题目三piece = str(model.predict([[12]])[0][0])[:5]print("预测12英寸披萨的价格为:" + piece)# 题目四data_x_test = [[8], [9], [11], [12], [16]]data_y_test = [[8.5], [11], [12], [15], [18]]data_x2_test = [[6], [11], [14], [18]]plt = run_plt()plt.plot(data_x, data_y, 'k.')plt.plot(data_x_test, data_y_test, 'r.')data_y2_test = model.predict(data_x2_test)plt.plot(data_x2_test, data_y2_test, '-')plt.show()# 计算模型准确率print(model.score(data_x_test, data_y_test))

结果图：

好啦，到这里这个实验就完成了，这是我第一次写博客，写的不好的地方请大家多多指教，对了，未来cust的学弟学妹们，你们的学长呕心沥血地写这个博客不是为了让你们“借鉴”的哦，学长的初衷是为了你们少走弯路哦！