add_fitted.gam: Add fitted values from a GAM to a data frame

In gavinsimpson/gratia: Graceful 'ggplot'-Based Graphics and Other Functions for GAMs Fitted Using 'mgcv'

Add fitted values from a GAM to a data frame

Description

Add fitted values from a GAM to a data frame

Usage

## S3 method for class 'gam'
add_fitted(data, model, value = ".fitted", type = "response", ...)

Arguments

data	a data frame containing values for the variables used to fit the model. Passed to stats::predict() as newdata.
model	a fitted model for which a stats::predict() method is available. S3 method dispatch is performed on the model argument.
value	character; the name of the variable in which model predictions will be stored.
type	character; the type of predictions to return. See mgcv::predict.gam() for options.
...	additional arguments passed to mgcv::predict.gam().

Value

A data frame (tibble) formed from data and predictions from model.

Examples

load_mgcv()
df <- data_sim("eg1", seed = 1)
df <- df[, c("y", "x0", "x1", "x2", "x3")]
m <- gam(y ~ s(x0) + s(x1) + s(x2) + s(x3), data = df, method = "REML")

# add fitted values to our data
add_fitted(df, m)

# with type = "terms" or "iterms"
add_fitted(df, m, type = "terms")

gavinsimpson/gratia documentation built on April 13, 2024, 10:56 p.m.