用R表示另一个表中的线性模型

时间:2017-06-10 10:57:58

标签: r predict

大家好我是R的新手,我需要知道如何在表格中使用预测功能。

我的第一张桌子上有一些天气变量(wind, temperature, pressure)和一个建筑物的consumption。我创建了一个直线模型:

mymodel<-lm(energyConsum ~ temperature + pressure+ wind, data=mytable)

我有第二个表格,其中包含天气预测,值为temperature, wind and pressure个变量,我想预测consumption

我知道我必须使用函数predict(),并设置mymodel,但我真的不知道如何创建一个包含所有预测的新列(按行)

欢呼声

1 个答案:

答案 0 :(得分:2)

我没有您的数据,但这是mtcars的示例。

您可以使用predict.lm的第二个参数newdata来预测data.frame。然后,您可以将这些结果分配给新列。

示例:

train <- mtcars[1:20, ]
test  <- mtcars[21:32,]

mymodel   <- lm(mpg ~ cyl + disp + hp, data=train)
test$pred <- predict(mymodel, test)