【求助】关于用R做prediction

2986

收藏 2011-02-06

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的两条直线

扫码加我拉你入群

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

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

全部回复

shenbaiseshatan

2011-2-6 09:03:06

可以尝试先将数据集分割

复制代码

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

扫码加我拉你入群

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

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

junuylia

2011-2-8 10:07:59

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

复制代码

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

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

扫码加我拉你入群

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

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

扫码加我 拉你入群

扫码加我 拉你入群

扫码加我 拉你入群

分享

扫码加好友，拉您进群

扫码加我拉你入群

扫码加我拉你入群

扫码加我拉你入群