使用R中的模型预测结果

Question

我正在尝试使用线性回归进行简单的预测 我有一个data.frame，其中一些项目缺少价格（因此注明NA）。 这显然不起作用：

#Simple LR
fit <- lm(Price ~ Par1 + Par2 + Par3, data=combi[!is.na(combi$Price),])
Prediction <- predict(fit,  data=combi[is.na(combi$Price),]), OOB=TRUE, type = "response")

我应该放什么而不是data=combi[is.na(combi$Price),])？

Answer 1

将data更改为newdata。查看?predict.lm以查看predict可以采用的参数。其他参数将被忽略。因此，在您的情况下，data（和OOB）会被忽略，默认情况下会返回对训练数据的预测。

Prediction <- predict(fit, newdata = combi[is.na(combi$Price),])

identical(predict(fit), predict(fit, data = combi[is.na(combi$Price),]))
## [1] TRUE