Add clustering algorithms

elixir-nx / scholar

Traditional machine learning on top of Nx

Apache License 2.0

395 stars 43 forks source link

Add clustering algorithms #135

Open josevalim opened 1 year ago

josevalim commented 1 year ago

[x] K-means
[x] Affinity Propagation
[ ] Mean Shift
[ ] Spectral clustering
[x] Hierarchical clustering
[x] DBSCAN
[ ] HDBSCAN
[ ] OPTICS
[ ] BIRCH

grevgeny commented 4 months ago

I'm interested in contributing to the project and would like to begin by implementing the Mean Shift algorithm.

josevalim commented 4 months ago

Please go ahead @grevgeny! You can see the PRs from other clustering algorithms, to get an idea how of these contributions usually work (and the project goals).

paulsullivanjr commented 2 months ago

I'm going to work on implementing BIRCH