twidlr: data.frame-based API for model and predict functons
@drsimonj here to introduce my latest tidy-modelling package for R, “twidlr”. twidlr wraps model and predict functions you already know and love with a consistent data.frame-based API!
All models wrapped by twidlr can be fit to data and used to make predictions as follows:
library(twidlr) fit <- model(data, formula, ...) predict(fit, data, ...)
datais a data.frame (or object that can be corced to one) and is required
formuladescribes the model to be fit
The APIs of model and predict functions in R are inconsistent and messy.
Some models like linear regression want a formula and data.frame:
lm(hp ~ ., mtcars)
Models like gradient-boosted decision trees want vectors and matrices:
library(xgboost) y <- mtcars$hp x <- as.matrix(mtcars[names(mtcars) != "hp"]) xgboost(x, y, nrounds = 5)
Models like generalized linear models want you to work. For example, to create interactions and dummy-coded variables:
library(glmnet) y <- iris$Petal.Length x <- model.matrix(Petal.Length ~ Sepal.Width * Sepal.Length + Species, iris) glmnet(x, y)
Some models like k-means don’t have a corresponding predict function:
fit <- kmeans(iris[1:120,-5], centers = 3) predict(fit, iris[121:150,]) ## Error in UseMethod("predict") : ## no applicable method for 'predict' applied to an object of class "kmeans"
Some predict functions are impure and return unexpected results. For example, linear discriminant analysis:
library(MASS) d <- iris fit <- lda(Species ~ ., d) table(predict(fit)$class) #> #> setosa versicolor virginica #> 50 49 51 d <- d[1:10,] table(predict(fit)$class) #> #> setosa versicolor virginica #> 10 0 0
Principal components is another example. You can fit the model with a complex formula, but can’t predict new data!
fit <- prcomp(~.*., mtcars[1:25, ]) # All 2-way interactions predict(fit, mtcars[26:32,]) ## Error in predict.prcomp(fit, mtcars[26:32, ]) : ## 'newdata' does not have named columns matching one or more of the original columns
twidlr helps to solve these problems by wrapping model and predict functions you already know and love with a consistent data.frame-based API!
Load twidlr and your favourite models can be fit to a data.frame with a formula and any additional arguments! To demonstrate, compare API to above:
library(twidlr) lm(mtcars, hp ~ .) xgboost(mtcars, hp ~ ., nrounds = 5) glmnet(iris, Petal.Length ~ Sepal.Width * Sepal.Length + Species)
What’s more, predictions can be made with all fitted models via
predict and a data.frame. This even works for models that don’t traditionally have a
library(twidlr) fit <- kmeans(iris[1:140,-5], centers = 3) predict(fit, iris[141:150,]) #>  3 3 3 3 3 3 3 3 3 3
Although useful in itself, a consistent data.frame-based API expands the capabilities of other tidy and data.frame-based packages like the tidyverse packages and pipelearner.
For the motivated, this demonstrates how to fit multiple models and compare their RMSE on new data. It’s streamlined because purrr’s map functions can exploit the consistent API for each model and
library(twidlr) library(purrr) train <- cars[ 1:40, ] test <- cars[41:50, ] f <- c("lm", "randomForest", "rpart") # Fit each model to training data and compute RMSE on test data rmse <- invoke_map(f, data = train, formula = speed ~ dist) %>% map(predict, data = test) %>% map_dbl(~ sqrt(mean((. - test$speed)^2))) setNames(rmse, f) #> lm randomForest rpart #> 3.832426 6.129539 6.034932
If you can’t see the value, try doing this without twidlr.
Take home messages
twidlr attempts to brings the follwing to modelling in R:
- a consistent and tidy model APIs
- pure and available predict functions
- the power of formula operators
- tidyverse philosophy (eg keep piping)
But twidlr is new, and needs your help to grow. So if your favourite model isn’t listed here, fork twidlr on GitHub and add it to help improve modelling in R! Advice for contributing can be found here.
Thanks already to Joran Elias and Mathew Ling for their contributions!
Thanks for reading and I hope this was useful for you.
For updates of recent blog posts, follow @drsimonj on Twitter, or email me at email@example.com to get in touch.
If you’d like the code that produced this blog, check out the blogR GitHub repository.