我正在尝试运行logit模型并绘制一些重要预测变量的概率曲线。我正在努力做到这一点 使用效果包。
df=data.frame(income=c(5,5,3,3,6,5),
won=c(0,0,1,1,1,0),
age=c(18,18,23,50,19,39),
home=c(0,0,1,0,0,1))
str(df)
md1 = glm(factor(won) ~ income + age + home,
data=df, family=binomial(link="logit"))
summary(md1)
plot(effect("income", md1), grid=TRUE)
但我想知道如何绘制图表,以便根据收入(或任何其他预测因子)显示获胜(响应)的概率。
但是,我想要做的是生成相同的图,y轴为赢,x轴为收入,但曲线显示年龄和家庭的概率(在不同的图中)。
在效果文档中没有看到如何执行此操作。救命啊!
感谢。
答案 0 :(得分:1)
如果我理解正确,您需要在模型中引入交互项。例如,
md1 <- glm(factor(won) ~ income*age + home,
data=df, family=binomial(link="logit"))
with(df, mean(age))
#[1] 27.83333
with(df, mean(age)-sd(age))
#[1] 14.33642
with(df, mean(age)+sd(age))
#[1] 41.33025
plot(effect("income*age", md1,
xlevels=list(income=0:10, age=c(14.3, 27.8, 41.3))), multiline=TRUE)
将向您展示不同年龄段的收入对获胜的影响。