0

I want to run kneighborregressor and kneighborclassifier on GPU. However, I cannot find out them in RAPIDS. Are these functions currently implemented or planned to be developed?

kou murakami
  • 11

1 Answers1

1

The underlying code is there for KNN in RAPIDS 0.10, but it does not yet have the Regressor and Classifier interfaces. You can use cuml KNN to find the nearest neighbors themselves, though. In 0.11, we will support these interfaces like in sklearn.

JohnZed
  • 36
  • 1