linkage.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2018-2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cuml/common/distance_type.hpp>
 
#include <raft/core/handle.hpp>
 
namespace ML {
namespace linkage {
 
void single_linkage(const raft::handle_t& handle,
                    const float* X,
                    int n_rows,
                    int n_cols,
                    size_t n_clusters,
                    ML::distance::DistanceType metric,
                    int* children,
                    int* labels,
                    bool use_knn = false,
                    int c        = 15);
 
};  // namespace linkage
};  // namespace ML