cupy / cupy/cupy

Equivalent of scipy.cluster.vq.kmeans

Open
#5,947 1 comment 2 reactions 0 assignees View on GitHub
cat:feature prio:medium
Dominant language
Python
Stars
12.3k
Forks
1.1k
Avg merge
1d 20h
Merged PRs (30d)
45

Description

### Description

The RAFT project (https://github.com/rapidsai/raft) has several building blocks for different algorithms, which include the k-means and single-linkage clustering algorithms, which could be exposed directly to cupy. I'm hoping we might be able to expose this in the form of a `pylibraft` artifact, which could expose either the cython or the Python in a similar way to `pylibcugraph`.

Contributor guide

Open the contributing guide

Research direction

Review the RAFT k-means and single-linkage building blocks and compare how pylibcugraph exposes similar functionality. Define how a pylibraft artifact would expose these algorithms to CuPy through Python or Cython; done means the requested clustering functionality is available through that integration.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.