Get startedGet started for free

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

View Course

Exercise instructions

  • Fit a GAM to the mpg data, modeling city.mpg as a sum of smooth functions of weight, length, and price, and also include the categorical terms fuel, drive, and style.
  • 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)
Edit and Run Code