-
Notifications
You must be signed in to change notification settings - Fork 662
[FEA] Support oob_prediction_ / oob_decision_function_ in Dask RandomForest #8501
Open
Description
Currently the Dask RandomForest throws NotImplementedError when the user attempts to access oob_prediction_ or oob_decision_function_ attributes. To implement these attributes, we need to perform allgather on rank-local OOB predictions along with partition-order metadata. See #8466 (comment) for the additional context.