【求助】关于用R做prediction

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

model=lm(salary~year+sex)sex 是 factor，分为male和female


我现在想要分别对male 和female做预测，并画出预测的直线，请问该怎么做啊？


也就是说，如果直接这样
new=seq（0,40,0.5）

plot(year,salary)
lines(new,predict(model,data.frame(year=new)))
出来的是一条直线，包含male 和female的，我想要做的是分开male和female的两条直线

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：Prediction predict Pred dic TIO 求助 Prediction

可以尝试先将数据集分割

1. dataf<-subset(da,sex=='F')
   
2. datam<-subset(da,sex=='M')

复制代码

然后再进行回归并把图画在一起

你可以给sex一个值1(male)或者0(female),你的model实际上是

1. salary=beta0+beta1*y+beta2*sex+error

复制代码

其中beta0是intercept, beta1 和beta2是coefficient，用这个来做回归。
如果sex=1, sal= (beta0+beta2)+beta1*y
如果sex=0, sal=beta0+beta1*y

这样你就可以通过一个model同时predict两个子模型了。当然楼上的方式也可以，但对比比较麻烦