机器学习作业二-线性回归预测销售额

题目如下： 

advertising.csv文件是某商品的广告推广费用（单位为元）和销售额数据（单位为千元），其中每行代表每一周的广告推广费用（包含微信、微博和其他类型三种广告费用）和销售额。若在未来的某两周，将各种广告投放金额按如下分配，请预测对应的商品销售额：

（1）微信：100，微博：100，其他类型：100

（2）微信：200，微博：100，其他类型：50

一、线性回归概念

对于因变量y：

如果它和自变量x呈现y=ax+b关系 称为一元线性。

如果y与多个因素有关。即 y=a1*x1+a2*x2+...  称为多元线性。

回归要做的就是根据已有x和y 找到a和b,拟合出这一条直线，预测新x所对应的y。

二、题目分析 

首先得到数据（使用pandas），然后画散点图（使用plt）观察三个因素对销量的影响。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression as lr

adv=pd.read_csv("advertising.csv")
#print(adv)

x=np.array(adv.wechat)
#x1=np.array(adv.weibo)
#x2=np.array(adv.others)
y=np.array(adv.sales)
plt.scatter(x,y)
plt.show()#得到wechat和sales的散点图
#发现只有wechat和sales有线性关系，所以是一元线性回归

  画出三个要素对应的散点图。发现只有微信部分和销售额有明显线性关系。

  

 参考文章