scikit-learn源码学习之cluster.mean_shift.estimate_bandwidth

最新推荐文章于 2025-03-21 10:14:42 发布

原创

最新推荐文章于 2025-03-21 10:14:42 发布 · 6.4k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#sklearn #聚类 #python #机器学习 #源码

本文深入探讨了scikit-learn库中mean-shift聚类算法的estimate_bandwidth函数，该函数用于自动计算聚类的bandwidth。在MeanShift未指定bandwidth参数时，此函数至关重要。内容包括源码解析和个人理解，欢迎讨论。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

继续我的源码学习之旅，这次是mean-shift聚类算法里面的estimate_bandwidth函数。
estimate_bandwidth函数用作于mean-shift算法估计带宽，如果MeanShift函数没有传入bandwidth参数，MeanShift会自动运行estimate_bandwidth，源码地址

def estimate_bandwidth(X, quantile=0.3, n_samples=None, random_state=0,
                       n_jobs=1):
    """Estimate the bandwidth to use with the mean-shift algorithm.

    That this function takes time at least quadratic in n_samples. For large
    datasets, it's wise to set that parameter to a small value.

    Parameters
    ----------
    X : array-like, shape=[n_samples, n_features]