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

Question

正在使用 Python 进行 SOM 实现吗？我想知道如何生成u矩阵

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 矩阵的实现。