Ejemplo n.º 1
0
def spectral_clustering(similarity_matrix, k):
    w, v = linalg.eig(
        similarity_matrix - np.diagflat(similarity_matrix.sum(axis=0)))
    centers, label = k_means(v[:, np.argsort(w)[-k:]], k)
    return label
Ejemplo n.º 2
0
 def test_k_means(self):
     center, label = k_means(np.matrix(data), 5)
     assert len(set(label)) == 5
Ejemplo n.º 3
0
 def test_k_means(self):
     center, label = k_means(np.matrix(data), 5)
     assert len(set(label)) == 5
Ejemplo n.º 4
0
def spectral_clustering(similarity_matrix, k):
    w, v = linalg.eig(similarity_matrix -
                      np.diagflat(similarity_matrix.sum(axis=0)))
    centers, label = k_means(v[:, np.argsort(w)[-k:]], k)
    return label