Como posso prever valores de novas entradas de um modelo linear em R?

21

Eu criei um modelo linear em R: mod = lm(train_y ~ train_x). Quero passar uma lista de Xs e obter seu Y previsto / estimado / previsto. Olhei predict(), mas acho que é para outra coisa, ou simplesmente não sei como usá-lo.

Suponho que, tomando os coeficientes do meu modelo, eu poderia manualmente plug-in as variáveis ​​test_x uma a uma e obter um Y previsto, mas acho que existe uma maneira mais eficiente de fazer isso.

Dolan Antenucci
fonte

Respostas:

20

Se você deseja os valores previstos para train_x= 1, 2 e 3, use predict(mod, data.frame(train_x = c(1, 2, 3))).

mark999
fonte
Eu estava deixando de fora o "train_x =" e adicionando isso resolveu meu problema! Muito obrigado
Dolan Antenucci