Добавление категориальных переменных в модель производительности автомобиля
Теперь вы включите категориальные переменные в модель. По своей природе категориальные переменные линейны, поэтому их следует моделировать как линейные члены.
Это упражнение является частью курса
Нелинейное моделирование с помощью обобщённых аддитивных моделей (GAM) в R
Инструкции к упражнению
- Обучите GAM на данных
mpg, моделируяcity.mpgкак сумму гладких функций отweight,lengthиprice, а также включите категориальные членыfuel,driveиstyle. - Используйте функцию
plot()для визуализации модели.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
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)