search

rapidsai / cuml

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

[QST] How to implement dictionary learning algorithm #5789

Open GraceZhuuu opened 7 months ago

GraceZhuuu commented 7 months ago

What is your question? I did not find the relevant algorithm for dictionary learning like sklearn. How can I use cuML to implement this algorithm?

dantegd commented 6 months ago

Thanks for the reques @GraceZhuuu! We currently do not have an implemented. That said, we have solvers available and most (all?) of the required primitives in cuML and RAFT to implement it. Contributions are always welcome, but otherwise we will consider implementing it in an upcoming version.