33. 常用聚类算法对比评估#

33.1. 介绍#

本周的课程讲解了近十种不同思路的聚类算法，最后一个挑战的目的是探索不同算法在不同形状数据集上的聚类效果和所需时间。

33.2. 知识点#

  • 算法对数据形状的适应性

  • 相同条件下的算法效率

33.3. 生成测试数据#

挑战前，需要先生成测试数据，我们使用 make_moons()，make_circles()，make_blobs() 生成 3 组数据。

Exercise 33.1

挑战：使用 sklearn 生成 3 组测试数据，并绘制出散点图（横排拼接子图）。

规定：

  • 为了方便后续实验，你需要对 3 组测试数据进行 Min-Max 归一化处理，使得横纵坐标都在 0-1 的范围之内。

  • 3 种方法对应的参数如下：

  • 三组数据均包含 100 个样本，且将随机数种子均设为 10。

  • moons 和 circles 数据添加 0.1 的噪声。

  • circles 数据内外圈间距比例因子设为 0.3。

  • blobs 数据拥有 2 个簇，且簇标准差设为 1.5。

  • 其余参数取默认值。

Min-Max 标准化同样是常用手段，其效果类似于区间缩放，可以将数值缩放到 0-1 之间。其公式为：

\[\hat x=\frac{x-x_{min}}{x_{max}-x_{min}}\]

其中，\(x_{max}\) 为样本数据的最大值，\(x_{min}\) 为样本数据的最小值，\(x_{max}-x_{min}\) 为极差。

import numpy as np
from sklearn import datasets
from matplotlib import pyplot as plt

%matplotlib inline

## 代码开始 ### (≈ 7-10 行代码)

## 代码结束 ###

期望输出

image

33.4. 聚类结果对比#

接下来，拟选用 sklearn 提供的几种常用的聚类方法在上面 3 个数据集上测试聚类效果，这些聚类方法有：

  • cluster.KMeans()

  • cluster.MiniBatchKMeans()

  • cluster.AffinityPropagation()

  • cluster.MeanShift()

  • cluster.SpectralClustering()

  • cluster.AgglomerativeClustering()

  • cluster.Birch()

  • cluster.DBSCAN()

Exercise 33.2

挑战：采用上述 8 个聚类方法分别在 moons, circles 和 blobs 上进行测试，并将聚类结果绘制成图。

规定：本次挑战不规定参数，请根据前面学习的内容使得聚类效果尽量更好。

from sklearn import cluster

## 代码开始 ### (> 10 行代码)

## 代码结束 ###

期望输出

image

上图中，我们一般只能得出同一种方法在不同数据集上聚类效果好坏的结论。由于不同方法之间的聚类效果取决于具体的参数，所以并不能准确得到谁好谁坏的结果，只能作为参考。

接下来，我们想测试不同方法在不同数量级的数据集上执行速度的快慢对比。这一次，我们仅使用 make_blobs() 生成团状测试数据。生成测试数据的函数如下：

def create_data(n):
    """
    参数:
    n -- 生成样本数量
    返回:
    blobs_data -- 样本数组
    """
    blobs_data, _ = datasets.make_blobs(
        n_samples=n, n_features=2, centers=2, cluster_std=1.5, random_state=10
    )
    return blobs_data

Exercise 33.3

挑战：使用 create_data(n) 生成不同规模的测试数据，并计算不同算法的聚类时间，且绘制对应的折线图。

规定：

  • 生成数据规模依次为：100, 200, 300, ···, 900, 1000 等 10 组。

  • 除指定 n_clusters=2 之外，其余参数均使用相关方法的默认参数。

提示：可以通过 time 模块记录算法运行时间，例如：

import time

t0 = time.time()
cluster_estimator.fit()
t1 = time.time()

则 t1-t0 即为 cluster_estimator.fit() 所花费时间。

import time

## 代码开始 ### (> 10 行代码)

## 代码结束 ###

期望输出

image

可以发现，Affinity Propagation，Mean Shift 以及 Spectral Clustering 花费时间上升速度较快，也是比较耗时的聚类算法。