如何在通过weka-python-wrapper进行聚类后确定每个实例的集群分配

Question

from weka.clusterers import Clusterer

clusterer = Clusterer(classname="weka.clusterers.SimpleKMeans", options=["-N", "6"])
clusterer.build_clusterer(data)

这是聚类

在此之后，我想知道每个实例的聚类分配。我们怎么做？

Answer 1

您可以使用cluster_instance(Instance)方法获取群集的从0开始的索引，或使用distribution_for_instance(Instance)方法获取群集分布：

for inst in data:
    cl = clusterer.cluster_instance(inst)
    dist = clusterer.distribution_for_instance(inst)
    print("cluster=" + str(cl) + ", distribution=" + str(dist))