Published by the Foundation for Open Access Statistics Editors-in-chief: Bettina Grün, Torsten Hothorn, Edzer Pebesma, Achim Zeileis    ISSN 1548-7660; CODEN JSSOBK
Authors: Marjolein Fokkema
Title: 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 shows how they can be fitted using function pre from R package pre, which derives PREs largely through the methodology of Friedman and Popescu (2008). The implementation and functionality of pre is described and illustrated through application on a dataset on the prediction of depression. Furthermore, accuracy and sparsity of pre is compared with that of single trees, random forests, lasso regression and the original RuleFit implementation of Friedman and Popescu (2008) in four benchmark datasets. Results indicate that pre derives ensembles with predictive accuracy similar to that of random forests, while using a smaller number of variables for prediction. Furthermore, pre provided better accuracy and sparsity than the original RuleFit implementation.

Page views:: 1237. Submitted: 2017-08-05. Published: 2020-03-16.
Paper: Fitting Prediction Rule Ensembles with R Package pre     Download PDF (Downloads: 275)
pre_1.0.0.tar.gz: R source package Download (Downloads: 20; 273KB) Replication materials Download (Downloads: 24; 91KB)

DOI: 10.18637/jss.v092.i12

This work is licensed under the licenses
Paper: Creative Commons Attribution 3.0 Unported License
Code: GNU General Public License (at least one of version 2 or version 3) or a GPL-compatible license.