FalkonML / falkon

Large-scale, multi-GPU capable, kernel solver
https://falkonml.github.io/falkon/
MIT License
181 stars 22 forks source link

v0.6 -- InCoreFalkon, CUDA LAUUM, Bug fixes #10

Closed Giodiro closed 4 years ago

Giodiro commented 4 years ago

Large pull request to incorporate several changes.

The driving change was the implementation of an in-core version of Falkon, suitable for smaller data analyses. Here the data is always kept inside the GPU, thus the model can train much faster. The result is the InCoreFalkon class.

LAUUM was improved to use a CUDA implementation for the inner-loop function.

Several bug fixes were also introduced, and edge-cases fixed.