使用效果包绘制logit概率

时间:2012-08-13 19:52:03

标签: r

我正在尝试运行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轴为收入,但曲线显示年龄和家庭的概率(在不同的图中)。

在效果文档中没有看到如何执行此操作。救命啊!

感谢。

1 个答案:

答案 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)

将向您展示不同年龄段的收入对获胜的影响。

相关问题