A proposal for supervised clustering with Dirichlet Process using labels

Billy Peralta, Luis Alberto Caro, Alvaro Soto

Abstract: Highlights•We propose to modify the prior process of clusters by considering label information.•We heuristically modify the prior process of clusters considering a Polya Urn model.•We test five real datasets with 30 random-holdout obtaining good performance in comparison to other alternatives.•We analyse five real datasets respect to MCMC learning process considering log-likelihood and number of clusters.•We recommend to this variant because it has better performance that clustering based on Dirichlet Process.