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: Thomas Grubinger, Achim Zeileis, Karl-Peter Pfeiffer
Title: evtree: Evolutionary Learning of Globally Optimal Classification and Regression Trees in R
Abstract: Commonly used classification and regression tree methods like the CART algorithm are recursive partitioning methods that build the model in a forward stepwise search. Although this approach is known to be an efficient heuristic, the results of recursive tree methods are only locally optimal, as splits are chosen to maximize homogeneity at the next step only. An alternative way to search over the parameter space of trees is to use global optimization methods like evolutionary algorithms. This paper describes the evtree package, which implements an evolutionary algorithm for learning globally optimal classification and regression trees in R. Computationally intensive tasks are fully computed in C++ while the partykit package is leveraged for representing the resulting trees in R, providing unified infrastructure for summaries, visualizations, and predictions. evtree is compared to the open-source CART implementation rpart, conditional inference trees (ctree), and the open-source C4.5 implementation J48. A benchmark study of predictive accuracy and complexity is carried out in which evtree achieved at least similar and most of the time better results compared to rpart, ctree, and J48. Furthermore, the usefulness of evtree in practice is illustrated in a textbook customer classification task.

Page views:: 5004. Submitted: 2012-07-05. Published: 2014-10-24.
Paper: evtree: Evolutionary Learning of Globally Optimal Classification and Regression Trees in R     Download PDF (Downloads: 3426)
evtree_1.0-0.tar.gz: R source package Download (Downloads: 275; 1MB)
v61i01.R: R example code from the paper Download (Downloads: 348; 17KB) Benchmark simulation results (in R binary format) and replication scripts Download (Downloads: 232; 243KB)

DOI: 10.18637/jss.v061.i01

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.