NVIDIA / NVIDIA/raft

[FEA] Expose make_monotonic in pylibraft

Open
#640 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

feature request good first issue python
Dominant language
Cuda
Stars
1k
Forks
251
Avg merge
1d 8h
Merged PRs (30d)
31

Description

cuML users would like estimators to accept non-consecutive target values outside of 0, n), which can occur in scenarios such as training on imbalanced datasets with shuffled data and cross-validation.

cuML has a make_monotonic utility to do this based on CuPy rawkernels, but this requires JIT overhead. We could instead use the make_monotonic primitive in RAFT, which was implemented for DBSCAN in cuML but moved to RAFT.

We can expose this primitive in pylibraft and use it in cuML to both avoid the performance overhead and clean up the cuML codebase.

In case it's relevant, there is also a separate request for renumbering for a possibly different use case.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by locating the existing RAFT make_monotonic primitive and the pylibraft binding entry points. Review how similar RAFT primitives are exposed and tested; done means cuML can access make_monotonic through pylibraft without the CuPy rawkernel JIT overhead.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, python
Domain
api, machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
42/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.