python 实现spectral clustering

最新推荐文章于 2024-08-15 10:04:41 发布

zhoucy163

最新推荐文章于 2024-08-15 10:04:41 发布

阅读量1.8k

点赞数 1

分类专栏： ML 图像处理和计算机视觉

本文链接：https://blog.csdn.net/jiao_zhoucy/article/details/26501927

版权

图像处理和计算机视觉同时被 2 个专栏收录

52 篇文章 1 订阅

订阅专栏

23 篇文章 0 订阅

订阅专栏

import numpy as np
import math
import sys
from scipy.cluster.vq import kmeans2

def 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,centroids
	return label;