python - 关于如何使用 python（在 SOM 上工作）绘制 U-MATRIX 的任何示例/想法？

Question

使用 Python 进行 SOM 实现？我想知道如何生成 u-matrix

标签： pythonplotsom

Answer 1

一种实现可以在SUSI包 ( pip3 install susi) 中找到。你可以这样使用它：

import susi
import matplotlib.pyplot as plt
from sklearn.datasets import make_blobs

# get data (replace this part with your data)
X, y = make_blobs(n_samples=100, n_features=2, centers=3)

# initialize and fit SOM
som = susi.SOMClustering()
som.fit(X)

u_matrix = som.get_u_matrix()
plt.imshow(np.squeeze(u_matrix), cmap="Greys")
plt.colorbar()
plt.show()

这导致了这个情节：

代码和绘图取自susi/SOMClustering.ipynb。你也可以在那里找到 u-matrix 的实现。