python 实现spectral clustering

来源:互联网 发布:淘宝追加评论期限 编辑:程序博客网 时间:2024/05/16 01:39
import numpy as npimport mathimport sysfrom scipy.cluster.vq import kmeans2def SpectralClustering(simi_matrix,cluster_num):N,N = np.shape(simi_matrix);DN = np.diag(1/np.sqrt(np.sum(simi_matrix,axis=1)));LapN = np.eye(N) - np.dot(np.dot(DN,simi_matrix),DN);U,s,V = np.linalg.svd(LapN,full_matrices=True);kerN = U[:,N-cluster_num+1:N];for i in range(N):kerN[i,:] = kerN[i,:] / np.linalg.norm(kerN[i,:]);centroids,label = kmeans2(kerN,cluster_num,iter=20);del DN,U,s,V,kerN,centroidsreturn label;

0 0