Multinomial probit Bayesian additive regression trees

Stat (Int Stat Inst). 2016;5(1):119-131. doi: 10.1002/sta4.110. Epub 2016 Apr 4.

Abstract

This article proposes multinomial probit Bayesian additive regression trees (MPBART) as a multinomial probit extension of BART - Bayesian additive regression trees. MPBART is flexible to allow inclusion of predictors that describe the observed units as well as the available choice alternatives. Through two simulation studies and four real data examples, we show that MPBART exhibits very good predictive performance in comparison to other discrete choice and multiclass classification methods. To implement MPBART, the R package mpbart is freely available from CRAN repositories.

Keywords: Bayesian methods; Classification; Machine learning; Statistical computing.