Introduction to Writing Functions in R
Richie Cotton
Data Evangelist at DataCamp
A linear model
lm(
response_var ~ explanatory_var1 + explanatory_var2,
data = dataset
)
A generalized additive model
library(mgcv)
gam(
response_var ~ s(explanatory_var1) + explanatory_var2,
data = dataset
)
predict_this <- data.frame(
explanatory_var1 = c("some", "values"),
explanatory_var2 = c("more", "values")
)
predicted_responses <- predict(model, predict_this, type = "response")
predict_this %>%
mutate(predicted_responses = predicted_responses)
Introduction to Writing Functions in R