Сравнение производительности алгоритмов Bisecting K-Means и обычного K-Means

Этот пример демонстрирует различия между алгоритмом обычного K-Means и алгоритмом Bisecting K-Means.

Хотя кластеризации K-Means отличаются при увеличении n_clusters, кластеризация Bisecting K-Means строится на основе предыдущих. В результате она имеет тенденцию создавать кластеры с более регулярной структурой большого масштаба. Это различие можно визуально наблюдать: для всех чисел кластеров существует разделительная линия, делящая облако данных пополам для BisectingKMeans, которая отсутствует для обычного K-Means.

# Authors: The scikit-learn developers
# SPDX-License-Identifier: BSD-3-Clause

import matplotlib.pyplot as plt

from sklearn.cluster import BisectingKMeans, KMeans
from sklearn.datasets import make_blobs

print(__doc__)


# Generate sample data
n_samples = 10000
random_state = 0

X, _ = make_blobs(n_samples=n_samples, centers=2, random_state=random_state)

# Number of cluster centers for KMeans and BisectingKMeans
n_clusters_list = [4, 8, 16]

# Algorithms to compare
clustering_algorithms = {
    "Bisecting K-Means": BisectingKMeans,
    "K-Means": KMeans,
}

# Make subplots for each variant
fig, axs = plt.subplots(
    len(clustering_algorithms), len(n_clusters_list), figsize=(12, 5)
)

axs = axs.T

for i, (algorithm_name, Algorithm) in enumerate(clustering_algorithms.items()):
    for j, n_clusters in enumerate(n_clusters_list):
        algo = Algorithm(n_clusters=n_clusters, random_state=random_state, n_init=3)
        algo.fit(X)
        centers = algo.cluster_centers_

        axs[j, i].scatter(X[:, 0], X[:, 1], s=10, c=algo.labels_)
        axs[j, i].scatter(centers[:, 0], centers[:, 1], c="r", s=20)

        axs[j, i].set_title(f"{algorithm_name} : {n_clusters} clusters")


# Hide x labels and tick labels for top plots and y ticks for right plots.
for ax in axs.flat:
    ax.label_outer()
    ax.set_xticks([])
    ax.set_yticks([])

plt.show()

Общее время выполнения скрипта: (0 минут 0.990 секунд)

Download Jupyter notebook: plot_bisect_kmeans.ipynb

Download Python source code: plot_bisect_kmeans.py

Download zipped: plot_bisect_kmeans.zip

Связанные примеры

Демонстрация предположений k-means

Сравнение алгоритмов кластеризации K-Means и MiniBatchKMeans

Пример инициализации K-Means++

Эмпирическая оценка влияния инициализации k-means