Classification Model Scoring

Accuracy

#include <raft/stats/accuracy.cuh>

namespace raft::stats

template<typename value_t, typename idx_t>
float accuracy(raft::resources const &handle, raft::device_vector_view<const value_t, idx_t> predictions, raft::device_vector_view<const value_t, idx_t> ref_predictions)

Compute accuracy of predictions. Useful for classification.

Template Parameters:

• value_t – data type for predictions (e.g., int for classification)

• idx_t – Index type of matrix extent.

Parameters:

• handle – [in] the raft handle.

• predictions – [in] array of predictions (GPU pointer).

• ref_predictions – [in] array of reference (ground-truth) predictions (GPU pointer).

Returns:

: Accuracy score in [0, 1]; higher is better.