Namespaces | Functions
metrics.hpp File Reference
#include <raft/distance/distance_types.hpp>
#include <cstdint>
Include dependency graph for metrics.hpp:

Go to the source code of this file.

Namespaces

 raft
 
 ML
 
 ML::Metrics
 
 ML::Metrics::Batched
 

Functions

float ML::Metrics::r2_score_py (const raft::handle_t &handle, float *y, float *y_hat, int n)
 
double ML::Metrics::r2_score_py (const raft::handle_t &handle, double *y, double *y_hat, int n)
 
double ML::Metrics::rand_index (const raft::handle_t &handle, double *y, double *y_hat, int n)
 
double ML::Metrics::silhouette_score (const raft::handle_t &handle, double *y, int nRows, int nCols, int *labels, int nLabels, double *silScores, raft::distance::DistanceType metric)
 
float ML::Metrics::Batched::silhouette_score (const raft::handle_t &handle, float *X, int n_rows, int n_cols, int *y, int n_labels, float *scores, int chunk, raft::distance::DistanceType metric)
 
double ML::Metrics::Batched::silhouette_score (const raft::handle_t &handle, double *X, int n_rows, int n_cols, int *y, int n_labels, double *scores, int chunk, raft::distance::DistanceType metric)
 
double ML::Metrics::kl_divergence (const raft::handle_t &handle, const double *y, const double *y_hat, int n)
 
float ML::Metrics::kl_divergence (const raft::handle_t &handle, const float *y, const float *y_hat, int n)
 
double ML::Metrics::entropy (const raft::handle_t &handle, const int *y, const int n, const int lower_class_range, const int upper_class_range)
 
double ML::Metrics::mutual_info_score (const raft::handle_t &handle, const int *y, const int *y_hat, const int n, const int lower_class_range, const int upper_class_range)
 
double ML::Metrics::homogeneity_score (const raft::handle_t &handle, const int *y, const int *y_hat, const int n, const int lower_class_range, const int upper_class_range)
 
double ML::Metrics::completeness_score (const raft::handle_t &handle, const int *y, const int *y_hat, const int n, const int lower_class_range, const int upper_class_range)
 
double ML::Metrics::v_measure (const raft::handle_t &handle, const int *y, const int *y_hat, const int n, const int lower_class_range, const int upper_class_range, double beta)
 
float ML::Metrics::accuracy_score_py (const raft::handle_t &handle, const int *predictions, const int *ref_predictions, int n)
 
void ML::Metrics::pairwise_distance (const raft::handle_t &handle, const double *x, const double *y, double *dist, int m, int n, int k, raft::distance::DistanceType metric, bool isRowMajor=true, double metric_arg=2.0)
 Calculates the ij pairwise distances between two input arrays of double type. More...
 
void ML::Metrics::pairwise_distance (const raft::handle_t &handle, const float *x, const float *y, float *dist, int m, int n, int k, raft::distance::DistanceType metric, bool isRowMajor=true, float metric_arg=2.0f)
 Calculates the ij pairwise distances between two input arrays of float type. More...
 
void ML::Metrics::pairwiseDistance_sparse (const raft::handle_t &handle, double *x, double *y, double *dist, int x_nrows, int y_nrows, int n_cols, int x_nnz, int y_nnz, int *x_indptr, int *y_indptr, int *x_indices, int *y_indices, raft::distance::DistanceType metric, float metric_arg)
 
void ML::Metrics::pairwiseDistance_sparse (const raft::handle_t &handle, float *x, float *y, float *dist, int x_nrows, int y_nrows, int n_cols, int x_nnz, int y_nnz, int *x_indptr, int *y_indptr, int *x_indices, int *y_indices, raft::distance::DistanceType metric, float metric_arg)
 
template<typename math_t , raft::distance::DistanceType distance_type>
double ML::Metrics::trustworthiness_score (const raft::handle_t &h, const math_t *X, math_t *X_embedded, int n, int m, int d, int n_neighbors, int batchSize=512)
 Compute the trustworthiness score. More...
 
double ML::Metrics::adjusted_rand_index (const raft::handle_t &handle, const int64_t *y, const int64_t *y_hat, const int64_t n)
 
double ML::Metrics::adjusted_rand_index (const raft::handle_t &handle, const int *y, const int *y_hat, const int n)