Risk bounds for embedded variable selection in classification trees - INRAE - Institut national de recherche pour l’agriculture, l’alimentation et l’environnement Accéder directement au contenu
Article Dans Une Revue IEEE Transactions on Information Theory Année : 2014

Risk bounds for embedded variable selection in classification trees

Résumé

The problems of model and variable selections for classification trees are jointly considered. A penalized criterion is proposed which explicitly takes into account the number of variables, and a risk bound inequality is provided for the tree classifier minimizing this criterion. This penalized criterion is compared to the one used during the pruning step of the CART algorithm. It is shown that the two criteria are similar under some specific margin assumptions. In practice, the tuning parameter of the CART penalty has to be calibrated by hold-out or cross-validation. A simulation study is performed to compare the form of the theoretical penalized criterion we propose with the form obtained after tuning the regularization parameter via cross-validation.

Dates et versions

hal-02636643 , version 1 (27-05-2020)

Identifiants

Citer

Servane Gey, Tristan Mary-Huard. Risk bounds for embedded variable selection in classification trees. IEEE Transactions on Information Theory, 2014, 60 (3), pp.1688-1699. ⟨10.1109/TIT.2014.2298874⟩. ⟨hal-02636643⟩
6 Consultations
0 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More