Adding categorical to the auto performance model
Now you'll include categorical variables in your model. Categories are inherently linear, so you'll model them as linear terms.
This exercise is part of the course
Nonlinear Modeling with Generalized Additive Models (GAMs) in R
Exercise instructions
- Fit a GAM to the
mpg
data, modelingcity.mpg
as a sum of smooth functions ofweight
,length
, andprice
, and also include the categorical termsfuel
,drive
, andstyle
. - Use the
plot()
function provided to visualize the model.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
library(mgcv)
# Fit the model
mod_city2 <- gam(city.mpg ~ s(weight) + s(length) + s(price) ___,
data = mpg, method = "REML")
# Plot the model
plot(mod_city2, all.terms = TRUE, pages = 1)