spectral_clustering.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include <raft/core/device_coo_matrix.hpp>
#include <raft/core/device_mdspan.hpp>
#include <raft/core/resources.hpp>
 
namespace ML {
namespace SpectralClustering {
 
struct params {
  int n_clusters;
  int n_components;
  int n_init;
  int n_neighbors;
  float eigen_tol;
  uint64_t seed;
};
 
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);
 
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);
 
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);
 
}  // namespace SpectralClustering
}  // namespace ML