pre: An R Package for Fitting Prediction Rule Ensembles

22 Jul 2017  ·  Marjolein Fokkema ·

Prediction rule ensembles (PREs) are sparse collections of rules, offering highly interpretable regression and classification models. This paper presents the R package pre, which derives PREs through the methodology of Friedman and Popescu (2008). The implementation and functionality of package pre is described and illustrated through application on a dataset on the prediction of depression. Furthermore, accuracy and sparsity of PREs is compared with that of single trees, random forest and lasso regression in four benchmark datasets. Results indicate that pre derives ensembles with predictive accuracy comparable to that of random forests, while using a smaller number of variables for prediction.

PDF Abstract

Categories


Computation Methodology

Datasets


  Add Datasets introduced or used in this paper