knn - rapids kneighborsclassifiers 和 kneighborsregressor 可以与 rapids 一起运行吗？

Question

我想在 GPU 上运行 kneighborregressor 和 kneighborclassifier。但是，我无法在 RAPIDS 中找到它们。这些功能目前是否已实施或计划开发？

Answer 1

RAPIDS 0.10 中有 KNN 的底层代码，但它还没有 Regressor 和 Classifier 接口。不过，您可以使用 cuml KNN 自己查找最近的邻居。在 0.11 中，我们将像在 sklearn 中一样支持这些接口。