Skip to content

[FEA] Support oob_prediction_ / oob_decision_function_ in Dask RandomForest #8501

Description

@chyunsu3

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions