¿Cómo puedo predecir valores a partir de nuevas entradas de un modelo lineal en R?

Question

Creé un modelo lineal en R: mod = lm(train_y ~ train_x) . Quiero pasarle una lista de X y obtener su Y predicho/estimado/pronosticado. Miré predict() , pero creo que es para otra cosa, o simplemente no sé cómo usarlo .

Supongo que al tomar los coeficientes de mi modelo, podría conectar manualmente las variables test_x una por una y obtener una Y predicha, pero supongo que hay una manera más eficiente de hacerlo.

Answer 1

Si desea los valores pronosticados para train_x = 1, 2 y 3, use predict(mod, data.frame(train_x = c(1, 2, 3))) .