__init__.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. """
  2. The :mod:`sklearn.cluster` module gathers popular unsupervised clustering
  3. algorithms.
  4. """
  5. from ._affinity_propagation import AffinityPropagation, affinity_propagation
  6. from ._agglomerative import (
  7. AgglomerativeClustering,
  8. FeatureAgglomeration,
  9. linkage_tree,
  10. ward_tree,
  11. )
  12. from ._bicluster import SpectralBiclustering, SpectralCoclustering
  13. from ._birch import Birch
  14. from ._bisect_k_means import BisectingKMeans
  15. from ._dbscan import DBSCAN, dbscan
  16. from ._hdbscan.hdbscan import HDBSCAN
  17. from ._kmeans import KMeans, MiniBatchKMeans, k_means, kmeans_plusplus
  18. from ._mean_shift import MeanShift, estimate_bandwidth, get_bin_seeds, mean_shift
  19. from ._optics import (
  20. OPTICS,
  21. cluster_optics_dbscan,
  22. cluster_optics_xi,
  23. compute_optics_graph,
  24. )
  25. from ._spectral import SpectralClustering, spectral_clustering
  26. __all__ = [
  27. "AffinityPropagation",
  28. "AgglomerativeClustering",
  29. "Birch",
  30. "DBSCAN",
  31. "OPTICS",
  32. "cluster_optics_dbscan",
  33. "cluster_optics_xi",
  34. "compute_optics_graph",
  35. "KMeans",
  36. "BisectingKMeans",
  37. "FeatureAgglomeration",
  38. "MeanShift",
  39. "MiniBatchKMeans",
  40. "SpectralClustering",
  41. "affinity_propagation",
  42. "dbscan",
  43. "estimate_bandwidth",
  44. "get_bin_seeds",
  45. "k_means",
  46. "kmeans_plusplus",
  47. "linkage_tree",
  48. "mean_shift",
  49. "spectral_clustering",
  50. "ward_tree",
  51. "SpectralBiclustering",
  52. "SpectralCoclustering",
  53. "HDBSCAN",
  54. ]