Add predictions to a data frame
Add predictions to a data frame
add_predictions(data, model, var = "pred", type = NULL) spread_predictions(data, ..., type = NULL) gather_predictions(data, ..., .pred = "pred", .model = "model", type = NULL)
data |
A data frame used to generate the predictions. |
model |
|
var |
The name of the output column, default value is |
type |
Prediction type, passed on to |
... |
|
.pred, .model |
The variable names used by |
A data frame. add_prediction adds a single new column,
with default name pred, to the input data.
spread_predictions adds one column for each model. gather_predictions
adds two columns .model and .pred, and repeats the input rows for each
model.
df <- tibble::tibble( x = sort(runif(100)), y = 5 * x + 0.5 * x ^ 2 + 3 + rnorm(length(x)) ) plot(df) m1 <- lm(y ~ x, data = df) grid <- data.frame(x = seq(0, 1, length = 10)) grid %>% add_predictions(m1) m2 <- lm(y ~ poly(x, 2), data = df) grid %>% spread_predictions(m1, m2) grid %>% gather_predictions(m1, m2)
Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.