Fix BaseRandomForestModel.

rapidsai · Feb 1, 2023 · 0c860b7 · 0c860b7
1 parent 8f271fb
commit 0c860b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/cuml/ensemble/randomforest_common.pyx b/python/cuml/ensemble/randomforest_common.pyx
@@ -290,6 +290,7 @@ class BaseRandomForestModel(Base):
                 y_m, cp.arange(self.num_classes, dtype=np.int32))
             if self.use_monotonic:
                 y_m, _ = make_monotonic(y_m)
+                y_m = input_to_cuml_array(y_m).array
 
         else:
             y_m, _, _, y_dtype = \