rapidsai / cuml

cuML - RAPIDS Machine Learning Library
https://docs.rapids.ai/api/cuml/stable/
Apache License 2.0
4.26k stars 535 forks source link

[FEA] Add new lanczos solver to UMAP for spectral initialization #6045

Open aamijar opened 3 months ago

aamijar commented 3 months ago

Documentation for future work:

So far, I've verified that the current spectral initialization gives random noise embeddings. One easy way to check this is to use a toy dataset from sklearn such as S-curve, comment out the UMAP code after spectral initialization and plot it.

Once the new solver is integrated the spectral initialization should be fixed.