Fitting Prediction Rule Ensembles with R Package pre
Abstract
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.
- Publication:
-
arXiv e-prints
- Pub Date:
- July 2017
- DOI:
- 10.48550/arXiv.1707.07149
- arXiv:
- arXiv:1707.07149
- Bibcode:
- 2017arXiv170707149F
- Keywords:
-
- Statistics - Computation;
- Statistics - Methodology
- E-Print:
- Journal of Statistical Software 92 (2020) 12 1-30