注意

RAFT 中的向量搜索和聚类算法正在迁移到一个专门用于向量搜索的新库，名为 cuVS。在此迁移期间，我们将继续支持 RAFT 中的向量搜索算法，但在 RAPIDS 24.06（六月）版本之后将不再更新它们。我们计划在 RAPIDS 24.10（十月）版本之前完成迁移，并在 24.12（十二月）版本中将它们完全从 RAFT 中移除。

分类模型评分

准确率

#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)

计算预测的准确率。用于分类。

模板参数:

• value_t – 预测的数据类型（例如，分类任务的 int）

• idx_t – 矩阵范围的索引类型。

参数:

• handle – [in] raft 句柄。

• predictions – [in] 预测数组（GPU 指针）。

• ref_predictions – [in] 参考（真实）预测数组（GPU 指针）。

返回值:

: 准确率得分，范围 [0, 1]；越高越好。