在阿里课程中看到的一个案例,使用医院线性回归预测电影的票房收入 问:大明电影公司投资拍摄了五部电影,并且整理了各影片的投资金额(百万元)和票房收入(百万元)。电影的投入和票房收入的数据见下面表格(如图)。接下来要拍一部投资2千万的电影,使用一元线性回归预测一下新电影的票房收入。
解答过程:1.python代码如下:
import matplotlib.pyplot as plt
def drawpltO:
plt.figure() plt.title( 'Cost and Income Of a Film')
plt.xlabe1( Cost(Mi11ion Yuan)')
plt.ylabel( Income(Million Yuan)')
plt.axis([0,25,0,60])
plt.grid(True)
X= [[6],[9],[12],[14],[16]]//将题中的x数据输入
y =[[9], [12],[29],[35],[59]]//y数据输入
drawplt()
plt.plot(X, y,'k.’)
plt.show()//生成线性图
2.生成的绘制的票房收入图为:
3.可以看出新电影票房收入,横轴数据在5-20之间;纵轴数据在10-60之间,呈现较为 明显的分布规律 4.开始使用线性回归预测电影票房收入,代码如下: import matplotlib.pyplot as plt
def drawplt():
plt.figure()
plt.title(' Cost and Income Of a Film')
plt. xlabe1(' Cost(Mi1lion Yuan)')
plt.ylabel(' Income (Million Yuan)')
plt.axis([0,25,0,60])
plt. grid(True)
X=[[6],[9],[12],[14],[16]]
y =[[9],[12],[29],[35],[59]]
model = linear_model. LinearRegression()
model.fit(X,y)
a=model.predict([[20]])
w=model.coef_
b=model.intercept_
print(a)
5.即可得到最终的结果69.95百万元
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。