首页 > 解决方案 > 图上的 sklearn 谱聚类生成具有未连接节点的子图

问题描述

我有一个需要聚类的 networkx 图。目前,我正在使用 sklearn 和光谱聚类。但是,下面的代码会生成未连接的子图(集群),即一个集群可以包含未连接到同一集群中的节点的节点。代码如下。

import networkx as nx  
from sklearn.cluster import SpectralClustering

G=nx.Graph()
G=nx.read_edgelist("edges.txt") # graph.txt contains the nodeedge list of your graph

adj_matrix = nx.to_numpy_matrix(G) 
   
clusters = SpectralClustering(affinity = 'precomputed',
                                  assign_labels="kmeans",
                                  random_state= 0,
                                  n_clusters= 4,
                                  n_init=100).fit_predict(adj_matrix)

有什么办法可以解决这个问题吗?如果这用另一种聚类方法做得更好,我很想知道。

标签: graphscikit-learnnetworkx

解决方案


推荐阅读