Cluster
=======

This page provides pylibraft class references for the publicly-exposed elements of the `pylibraft.cluster` package.

.. role:: py(code)
   :language: python
   :class: highlight

KMeans
######

.. autoclass:: pylibraft.cluster.kmeans.KMeansParams
    :members:

.. autofunction:: pylibraft.cluster.kmeans.fit

.. autofunction:: pylibraft.cluster.kmeans.cluster_cost

.. autofunction:: pylibraft.cluster.kmeans.compute_new_centroids
