#include <raft/core/handle.hpp>
#include <rmm/device_uvector.hpp>
#include <cuvs/distance/distance.hpp>
#include <cstddef>

Include dependency graph for hdbscan.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	ML::HDBSCAN::Common::CondensedHierarchy< value_idx, value_t >

class	ML::HDBSCAN::Common::RobustSingleLinkageParams

class	ML::HDBSCAN::Common::HDBSCANParams

class	ML::HDBSCAN::Common::robust_single_linkage_output< value_idx, value_t >

class	ML::HDBSCAN::Common::hdbscan_output< value_idx, value_t >

class	ML::HDBSCAN::Common::PredictionData< value_idx, value_t >

Namespaces
	ML

	ML::HDBSCAN

	ML::HDBSCAN::Common

	ML::HDBSCAN::HELPER

Enumerations
enum	ML::HDBSCAN::Common::CLUSTER_SELECTION_METHOD { ML::HDBSCAN::Common::EOM = 0 , ML::HDBSCAN::Common::LEAF = 1 }

Functions
void	ML::HDBSCAN::Common::generate_prediction_data (const raft::handle_t &handle, CondensedHierarchy< int, float > &condensed_tree, int labels, int inverse_label_map, int n_selected_clusters, PredictionData< int, float > &prediction_data)

void	ML::hdbscan (const raft::handle_t &handle, const float X, size_t m, size_t n, cuvs::distance::DistanceType metric, HDBSCAN::Common::HDBSCANParams &params, HDBSCAN::Common::hdbscan_output< int, float > &out, float core_dists)

void	ML::build_condensed_hierarchy (const raft::handle_t &handle, const int children, const float delta, const int *sizes, int min_cluster_size, int n_leaves, HDBSCAN::Common::CondensedHierarchy< int, float > &condensed_tree)

void	ML::_extract_clusters (const raft::handle_t &handle, size_t n_leaves, int n_edges, int parents, int children, float lambdas, int sizes, int labels, float probabilities, HDBSCAN::Common::CLUSTER_SELECTION_METHOD cluster_selection_method, bool allow_single_cluster, int max_cluster_size, float cluster_selection_epsilon)

void	ML::compute_all_points_membership_vectors (const raft::handle_t &handle, HDBSCAN::Common::CondensedHierarchy< int, float > &condensed_tree, HDBSCAN::Common::PredictionData< int, float > &prediction_data, const float X, cuvs::distance::DistanceType metric, float membership_vec, size_t batch_size=4096)

void	ML::compute_membership_vector (const raft::handle_t &handle, HDBSCAN::Common::CondensedHierarchy< int, float > &condensed_tree, HDBSCAN::Common::PredictionData< int, float > &prediction_data, const float X, const float points_to_predict, size_t n_prediction_points, int min_samples, cuvs::distance::DistanceType metric, float *membership_vec, size_t batch_size=4096)

void	ML::out_of_sample_predict (const raft::handle_t &handle, HDBSCAN::Common::CondensedHierarchy< int, float > &condensed_tree, HDBSCAN::Common::PredictionData< int, float > &prediction_data, const float X, int labels, const float points_to_predict, size_t n_prediction_points, cuvs::distance::DistanceType metric, int min_samples, int out_labels, float *out_probabilities)

void	ML::HDBSCAN::HELPER::compute_core_dists (const raft::handle_t &handle, const float X, float core_dists, size_t m, size_t n, cuvs::distance::DistanceType metric, int min_samples)
	Compute the core distances for each point in the training matrix. More...

void	ML::HDBSCAN::HELPER::compute_inverse_label_map (const raft::handle_t &handle, HDBSCAN::Common::CondensedHierarchy< int, float > &condensed_tree, size_t n_leaves, HDBSCAN::Common::CLUSTER_SELECTION_METHOD cluster_selection_method, rmm::device_uvector< int > &inverse_label_map, bool allow_single_cluster, int max_cluster_size, float cluster_selection_epsilon)
	Compute the map from final, normalize labels to the labels in the CondensedHierarchy. More...

Classes

Namespaces

Enumerations

Functions