谱聚类python代码_python中的谱聚类图

没有太多光谱聚类的经验，只是按照文档进行(结果请跳到最后！)以下内容：

代码：import numpy as np

import networkx as nx

from sklearn.cluster import SpectralClustering

from sklearn import metrics

np.random.seed(1)

# Get your mentioned graph

G = nx.karate_club_graph()

# Get ground-truth: club-labels -> transform to 0/1 np-array

# (possible overcomplicated networkx usage here)

gt_dict = nx.get_node_attributes(G, 'club')

gt = [gt_dict[i] for i in G.nodes()]

gt = np.array([0 if i == 'Mr. Hi' else 1 for i in gt])

# Get adjacency-matrix as numpy-array

adj_mat = nx.to_numpy_matrix(G)

print('ground truth')

print(gt)

# Cluster

sc = SpectralClustering(2, affinity='precomputed', n_init=100)

sc.fit(adj_mat)

# Compare ground-t