期刊
JOURNAL OF COMPUTATIONAL AND GRAPHICAL STATISTICS
卷 20, 期 1, 页码 260-278出版社
AMER STATISTICAL ASSOC
DOI: 10.1198/jcgs.2011.09066
关键词
Clustering; Nonparametric Bayes; Variable selection
资金
- NIH [1R01CA75981]
- FONDECYT [1060729]
- Laboratorio de Analisis Estocastico [PBCT-ACT13]
We propose a probability model for random partitions in the presence of covariates. In other words, we develop a model-based clustering algorithm that exploits available covariates. The motivating application is predicting time to progression for patients in a breast cancer trial. We proceed by reporting a weighted average of the responses of clusters of earlier patients. The weights should be determined by the similarity of the new patient's covariate with the covariates of patients in each cluster. We achieve the desired inference by defining a random partition model that includes a regression on covariates. Patients with similar covariates are a priori more likely to be clustered together. Posterior predictive inference in this model formalizes the desired prediction. We build on product partition models (PPM). We define an extension of the PPM to include a regression on covariates by including in the cohesion function a new factor that increases the probability of experimental units with similar covariates to be included in the same cluster. We discuss implementations suitable for any combination of continuous, categorical, count, and ordinal covariates. An implementation of the proposed model as R-package is available for download.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据