python - 如何从 PyMC3 中的 Dirichlet 过程中提取集群 ID 以获取分组数据？

Question

我正在使用 PyMC3 对我的分组数据进行聚类。基本上，我有 g 个向量，并且想将 g 个向量聚集成 m 个簇。但是，我有两个问题。

第一个是，PyMC3 似乎只能处理一维数据而不能处理向量。第二个问题是，我不知道如何提取原始数据的集群 ID。我确实提取了组件的数量（k）和相应的权重。但是我无法提取指示每个点属于哪个集群的 id。

欢迎任何想法或意见！

Answer 1

如果我理解正确，您正在尝试提取数据点属于哪个类别（1 到 k）。然而，狄利克雷随机变量只产生一个概率向量。这应该用作分类 RV 的先验，当从中采样时，它将产生一个编号的类别。