[BUG] RandomForestClassifier No attribute predict_model_on_cpu when it should ? #5008
Labels
? - Needs Triage
Need team to review and classify
bug
Something isn't working
Multi-GPU
Issues & PRs related to multi-GPU functionality
Describe the bug
Hello Nvidia,
I am encountering an issue during my usage of
from cuml.dask.ensemble import RandomForestClassifier
when aftethe fit operation I am trying to leverage the attributepredict_model_on_cpu
On the instalaltion it seems to exists when I am calling the help command
But when I want to use it , the attribute doesn't exist
Steps/Code to reproduce bug
Locally having parquet files to read with dask cudf ,
setup data one column that can be seen as a target and other column are features in float32
Overall there is the code doing the data prep
Expected behavior
Manage to make prediction from a CPU or break for memory issue but not founding the function seems not right
Environment details (please complete the following information):
starting from the AWS AMI ami-031863ae88fab9aeb
Additional context
The text was updated successfully, but these errors were encountered: