ML::SpectralClustering Namespace Reference

Classes
struct	params
	Spectral clustering parameters. More...

Functions
void	fit_predict (raft::resources const &handle, params config, raft::device_matrix_view< float, int, raft::row_major > dataset, raft::device_vector_view< int, int > labels)
	Perform spectral clustering on input dataset by constructing a k-nearest neighbors graph. More...
void	fit_predict (raft::resources const &handle, params config, raft::device_coo_matrix_view< float, int, int, int > connectivity_graph, raft::device_vector_view< int, int > labels)
	Perform spectral clustering on a precomputed connectivity graph using COO sparse matrix view. More...
void	fit_predict (raft::resources const &handle, params config, raft::device_vector_view< int, int > rows, raft::device_vector_view< int, int > cols, raft::device_vector_view< float, int > vals, raft::device_vector_view< int, int > labels)
	Perform spectral clustering on a precomputed connectivity graph using separate vector views for COO components. More...

Function Documentation

fit_predict() [1/3]

void ML::SpectralClustering::fit_predict	(	raft::resources const &	handle,
		params	config,
		raft::device_coo_matrix_view< float, int, int, int >	connectivity_graph,
		raft::device_vector_view< int, int >	labels
	)

Perform spectral clustering on a precomputed connectivity graph using COO sparse matrix view.

Parameters

[in]	handle	cuML resources handle
[in]	config	Parameters for spectral clustering
[in]	connectivity_graph	COO sparse matrix view of the connectivity graph
[out]	labels	Cluster labels for each sample

fit_predict() [2/3]

void ML::SpectralClustering::fit_predict	(	raft::resources const &	handle,
		params	config,
		raft::device_matrix_view< float, int, raft::row_major >	dataset,
		raft::device_vector_view< int, int >	labels
	)

Perform spectral clustering on input dataset by constructing a k-nearest neighbors graph.

Parameters

[in]	handle	cuML resources handle
[in]	config	Parameters for spectral clustering
[in]	dataset	Input dataset (row-major)
[out]	labels	Cluster labels for each sample

fit_predict() [3/3]

void ML::SpectralClustering::fit_predict	(	raft::resources const &	handle,
		params	config,
		raft::device_vector_view< int, int >	rows,
		raft::device_vector_view< int, int >	cols,
		raft::device_vector_view< float, int >	vals,
		raft::device_vector_view< int, int >	labels
	)

Perform spectral clustering on a precomputed connectivity graph using separate vector views for COO components.

Parameters

[in]	handle	cuML resources handle
[in]	config	Parameters for spectral clustering
[in]	rows	Row indices of the COO sparse matrix
[in]	cols	Column indices of the COO sparse matrix
[in]	vals	Values of the COO sparse matrix
[out]	labels	Cluster labels for each sample