我们如何使用自组织地图聚类为数据分配标签? (minisom包)python

时间:2019-08-29 11:20:19

标签: python som

我正在使用minisom软件包进行som集群。

train_batch()之后,如何使用训练有素的模型为数据点分配群集编号?

1 个答案:

答案 0 :(得分:0)

我可以为您提供SUSI软件包的代码,其中还包括一个用于集群化和可视化的不受监督的SOM 。您可以使用pip3 install susi安装它。这是一个代码示例(您可以在存储库的示例中找到更多示例):

import susi

# TODO get your data X here
X = ...

# train your SOM
som = susi.SOMClustering()
som.fit(X)

# get clusters
clusters = som.get_clusters(X)

# get u-matrix
u_matrix = som.get_u_matrix()