如何将预测列添加到数据框

时间:2015-05-19 15:37:43

标签: r dataframe logistic-regression

我使用以下代码来预测x的值。我想将预测值/列添加到数据框中。如何添加?

logit.fit<- glm(x~.-y, family = binomial(logit),data = data$train) 
logit.preds <- predict(logit.fit,newdata=data$val,type="response") 
logit.pred <- prediction(logit.preds,data$val$x) 
logit.perf <- performance(logit.pred,"tpr","fpr")

data $ val是验证集,数据$ train是训练集。

1 个答案:

答案 0 :(得分:1)

以下内容应该是您正在寻找的内容

install.packages("broom")
install.packages("ggplot2")

iris$y <- (iris$Species=="setosa")*1
logit.fit<- glm(y ~ Sepal.Length+Sepal.Width+Petal.Length, 
            family = binomial(logit), data = iris) 

View(ggplot2::fortify(logit.fit))
View(broom::augment(logit.fit))