arXiv · 1810.00029
Minimization of Gini impurity via connections with the k-means problem
Abstract
The Gini impurity is one of the measures used to select attribute in Decision Trees/Random Forest construction. In this note we discuss connections between the problem of computing the partition with minimum Weighted Gini impurity and the $k$-means clustering problem. Based on these connections we show that the computation of the partition with minimum Weighted Gini is a NP-Complete problem and we also discuss how to obtain new algorithms with provable approximation for the Gini Minimization problem.
Explore related subjects
Keep this discovery
Eduardo Sany Laber, Lucas Murtinho. 2018-09-28. Minimization of Gini impurity via connections with the k-means problem. https://arxiv.org/abs/1810.00029
Cite the original work for its findings. Save a collection to share your selection of sources.