如何使用python(在SOM上)绘制U-MATRIX的任何示例/想法?

时间:2019-01-21 13:01:01

标签: python plot som

使用Python进行SOM实现?我想知道如何生成u-矩阵

1 个答案:

答案 0 :(得分:0)

可以在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()

这将导致该图:

U-Matrix of SOM

代码和图解取自susi/SOMClustering.ipynb。您也可以在其中找到u-matrix的实现。