票房线性回归 Linear regression of film box office
【摘要】 1 数据统计与预览library(raster)a = read.csv("film.csv", header = T) ##读入数据,不要将字符串视为factorsummary(a) ##查看数据基本情况,检查缺失情况a = n...
1 数据统计与预览
library(raster)
a = read.csv("film.csv", header = T) ##读入数据,不要将字符串视为factor
summary(a) ##查看数据基本情况,检查缺失情况
a = na.omit(a) ##将数据集里全部NA所在行删除
film_high = a[which.max(a$boxoffice),] ##查看票房最高纪录
film_low = a[which.min(a$boxoffice),] ##查看票房最低纪录
2 数据预览
library(wordcloud)
col<-brewer.pal(8,"Accent")
h<-hist(a$boxoffice,breaks=12,col=col)
xfit <-seq(min(a$boxoffice),max(a$boxoffice),length=40)
yfit <-dnorm(xfit,mean=mean(a$boxoffice),sd=sd(a$boxoffice)) #dnorm返回的是正态分布概率函数值
yfit <- yfit*diff(h$mids[1:2]) *length(a$boxoffice)
lines(xfit,yfit,col='red',lwd=2)
summary(a$boxoffice)
数据右偏分布数据,对y取对数处理,左偏数据开方处理
3 分箱处理数据与预览
##票房直方图
boxbar<-as.data.frame(table(cut(a$boxoffice,breaks=c(1000*(1:5),7000,10000,50000,100000,130000),
labels=c('1-2千万','2-3千万','3-4千万','4-5千万',
'5-7千万','7千万-1亿','1-5亿','5-10亿','大于10亿'))))
barplot(boxbar$Freq,names.arg=boxbar$Var1,col="dodgerblue",xlab="票房(万元)",ylab="频数")
as.data.frame(table(a$genre))
boxplot(log(boxoffice)~genre,xlab="电影类型",
data=a,col="indianred1",ylab="对数票房",varwidth=T) ##电影类型箱线图
as.data.frame(table(a$month)) ##上映月份统计
boxplot(log(boxoffice)~month,xlab="月份",
data=a,col="dodgerblue",ylab="对数票房",varwidth=T)
```
4 完成数据清理函数
dataclean<-function(a){
a$dangqi<-cut(a$month,breaks=c(0,2,4,6,9,11),labels=c('贺岁档','普通档','黄金1档','暑期档','黄金2档'))
a$dangqi[is.na(a$dangqi)]<-'贺岁档'
a$distribute<-factor(a$distribute,levels=c('C','G','L','S'),
labels=c('联合发行','国有宣发','小私营公司','明星私企'))
a$reality<-factor(a$reality,levels=c(0,1),labels=c('否','是'))
a$story<-factor(a$story,levels=c('N','S'),labels=c('原创','虚构'))
a$remake<-factor(a$remake,levels=c(0,1),labels=c('否','是'))
a$sequel<-factor(a$sequel,levels=c(0,1),labels=c('否','是'))
a$导演得奖情况<-factor(a$导演得奖情况,levels=c(0,1),labels=c('否','是'))
a$导演是否转型<-factor(a$导演是否转型,levels=c(0,1),labels=c('否','是'))
a$导演年代[a$导演年代<50]="50以下" ##将30和40年代合并为50以下
a$导演年代=factor(a$导演年代,c("50以下","50","60","70","80")) ##更改因子水平顺序
return(a)
a<-dataclean(a)
table(a$dangqi) ##上映档期统计
boxplot(log(boxoffice)~dangqi,xlab="档期",
data=a,col="yellow",ylab="对数票房",varwidth=T) ##上映档期箱线图
table(a$distribute) ##宣发公司统计
boxplot(log(boxoffice)~distribute,xlab="宣发公司",
data=a,col="dodgerblue",ylab="对数票房",varwidth=T) ##宣发公司箱线图
par(mfrow=c(2,2)) ##输出2行2列图形
boxplot(log(boxoffice)~reality,data=a,col=c("dodgerblue"),
ylab="对数票房",xlab="是否真实",varwidth=T) ##是否真实
boxplot(log(boxoffice)~story,data=a,col=c("indianred1"),
ylab="",xlab="是否改编",varwidth=T) ##是否原创
boxplot(log(boxoffice)~remake,data=a,col=c("lightgreen"),
ylab="",xlab="是否翻拍",varwidth=T) ##是否翻拍
boxplot(log(boxoffice)~sequel,data=a,col=c("purple"),
ylab="",xlab="是否有续集",varwidth=T) ##是否续集
par(mfrow=c(1,2)) ##输出1行2列图形
boxplot(log(boxoffice)~导演得奖情况,data=a,col=c("indianred1"),
ylab="对数票房",xlab="",main="导演获奖 vs 票房",varwidth=T) ##导演是否获奖
boxplot(log(boxoffice)~导演是否转型,data=a,col=c("purple"),
ylab="",xlab="",main="导演转型 vs 票房",varwidth=T) ##导演是否转型
par(mfrow=c(1,1))
## 在比较导演年代时,我建议把30和40的统计归为50年代以下,这样不需要删样本
boxplot(log(boxoffice)~导演年代,data=a,col=c("orange"), ##不同年代导演的区别
ylab="对数票房",xlab="导 演 年 代",varwidth=T)
table(a$导演年代)
summary(a$boxoffice)
lm=lm(log(boxoffice)~genre+as.factor(year)+runtime+dangqi+distribute
+reality+story+remake+sequel+导演得奖情况+导演是否转型,data=a) ##拟合线性模型
summary(lm) ##给出估计结果
lm_s<-step(lm,direction="both")
summary(lm_s)
plot(resid(lm_s)~predict(lm_s))
abline(h=0)
sum(resid(lm_s)^2)/sum(predict(lm_s))
5 建模与评估
lm=lm(log(boxoffice)~genre+as.factor(year)+runtime+dangqi+distribute
+reality+story+remake+sequel+导演得奖情况+导演是否转型,data=a) ##拟合线性模型
summary(lm) ##给出估计结果
lm_s<-step(lm,direction="both")
summary(lm_s)
plot(resid(lm_s)~predict(lm_s))
abline(h=0)
sum(resid(lm_s)^2)/sum(predict(lm_s))
残差占比
6 残差分析
residplot=function(fit,nbreak=10){
z=rstandard(fit)
hist(z,breaks=nbreak,freq=F,xlab="Resuduals",main="Distribution of Errors")
curve(dt(x,1),
add=T,col="blue",lwd=2)
curve(dnorm(x,mean =mean(z),sd=sd(z)),add=T,col="green",lwd=2)
legend("topright",legend=c("Normal Curve","t curve"),lty=1:2,col=c("green","blue"),cex=0.8)
}
qqnorm(lm_s$residuals)
qqline(lm_s$residuals)
residplot(lm_s)
残差的结果符合正态性检验
7预测结果
library(car)
vif(lm_s) #viff方差膨胀系数,用于检测多重共线性
##预测第一部电影的票房
b = read.csv("predict.csv", header = T)
b<-dataclean(b)
bb = exp(predict(lm,b))
bb
【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)