Skip to content

Update Dask RF to use the new distributed algo - #8466

Open
chyunsu3 wants to merge 3 commits into
NVIDIA:mainfrom
chyunsu3:distributed_rf_dask
Open

Update Dask RF to use the new distributed algo#8466
chyunsu3 wants to merge 3 commits into
NVIDIA:mainfrom
chyunsu3:distributed_rf_dask

Conversation

@chyunsu3

@chyunsu3 chyunsu3 commented Aug 11, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor
  • Wire Dask RF training to RAFT communicators
  • Retain one fitted worker model as the canonical model for prediction and get_combined_model().
  • Remove obsolete components: per-worker n_estimator, ignore_empty_partitions, partial inference, etc
  • Update terminology: FIL -> nvForest
  • Use global n_rows when constraining certain hyperparameters
  • Update pytests

@copy-pr-bot

copy-pr-bot Bot commented Aug 11, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@github-actions github-actions Bot added Cython / Python Cython or Python issue CMake CUDA/C++ labels Aug 11, 2026
@chyunsu3
chyunsu3 marked this pull request as ready for review August 11, 2026 17:08
@chyunsu3
chyunsu3 requested a review from a team as a code owner August 11, 2026 17:08
@chyunsu3
chyunsu3 requested a review from viclafargue August 11, 2026 17:08
@josephine-wolf-oberholtzer josephine-wolf-oberholtzer added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Aug 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@viclafargue viclafargue Awaiting requested review from viclafargue viclafargue is a code owner automatically assigned from NVIDIA/cuml-python-codeowners

At least 1 approving review is required to merge this pull request.

Assignees

@chyunsu3 chyunsu3

Labels

CMake CUDA/C++ Cython / Python Cython or Python issue improvement Improvement / enhancement to an existing function non-breaking Non-breaking change

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@chyunsu3 @josephine-wolf-oberholtzer