譜協聚類算法的一個示例?

注意 單擊此處下載完整的示例代碼，或通過Binder在瀏覽器中運行此示例

有關列子在模塊sklearn.cluster.biclusterh中。

此示例演示如何使用譜協聚類算法(pectral Co-Clustering algorithm)生成數據集和雙光澤數據集。

數據集是使用make_biluster函數生成的，該函數創建一個小值矩陣和具有較大值的集群。然后對行和列進行洗牌，并將其傳遞給譜協聚類算法。重新排列洗牌矩陣，使集群連續顯示，該算法明確的表明地找到了集群。

# 輸出
# consensus score: 1.000

print(__doc__)

# Author: Kemal Eren <kemal@kemaleren.com>
# License: BSD 3 clause

import numpy as np
from matplotlib import pyplot as plt

from sklearn.datasets import make_biclusters
from sklearn.cluster import SpectralCoclustering
from sklearn.metrics import consensus_score

data, rows, columns = make_biclusters(
    shape=(300, 300), n_clusters=5, noise=5,
    shuffle=False, random_state=0)

plt.matshow(data, cmap=plt.cm.Blues)
plt.title("Original dataset")

# shuffle clusters
rng = np.random.RandomState(0)
row_idx = rng.permutation(data.shape[0])
col_idx = rng.permutation(data.shape[1])
data = data[row_idx][:, col_idx]

plt.matshow(data, cmap=plt.cm.Blues)
plt.title("Shuffled dataset")

model = SpectralCoclustering(n_clusters=5, random_state=0)
model.fit(data)
score = consensus_score(model.biclusters_,
                        (rows[:, row_idx], columns[:, col_idx]))

print("consensus score: {:.3f}".format(score))

fit_data = data[np.argsort(model.row_labels_)]
fit_data = fit_data[:, np.argsort(model.column_labels_)]

plt.matshow(fit_data, cmap=plt.cm.Blues)
plt.title("After biclustering; rearranged to show biclusters")

plt.show()

腳本的總運行時間：（0分鐘0.292秒）