similarity.py 文件源码

def spectral_clustering(S, X, config):
    '''
    Computes spectral clustering from an input similarity matrix.
    Returns the labels associated with the clustering.
    '''
    from sklearn.cluster import SpectralClustering

    nk = int(config["n_clusters"])
    clf = SpectralClustering(affinity='cosine', n_clusters=nk)
    return clf.fit_predict(X)