kmeans_mg.hpp File Reference

#include <cuml/cluster/kmeans.hpp>

Include dependency graph for kmeans_mg.hpp:

Go to the source code of this file.

Namespaces
	raft
	ML
	ML::kmeans
	ML::kmeans::opg

Functions
void	ML::kmeans::opg::fit (const raft::resources &handle, const KMeansParams &params, const float *X, int n_samples, int n_features, const float *sample_weight, float *centroids, float &inertia, int &n_iter)
	Compute k-means clustering. More...
void	ML::kmeans::opg::fit (const raft::resources &handle, const KMeansParams &params, const double *X, int n_samples, int n_features, const double *sample_weight, double *centroids, double &inertia, int &n_iter)
void	ML::kmeans::opg::fit (const raft::resources &handle, const KMeansParams &params, const float *X, int64_t n_samples, int64_t n_features, const float *sample_weight, float *centroids, float &inertia, int64_t &n_iter)
void	ML::kmeans::opg::fit (const raft::resources &handle, const KMeansParams &params, const double *X, int64_t n_samples, int64_t n_features, const double *sample_weight, double *centroids, double &inertia, int64_t &n_iter)