search

jdrusso / msm_we

History-augmented Markov analysis of weighted ensemble trajectories.
https://msm-we.readthedocs.io
MIT License
7 stars 7 forks source link

Implement ray + joblib for accelerated sklearn #33

Closed jdrusso closed 1 year ago

jdrusso commented 2 years ago

Ray provides a register_ray() function that allows you to use it as a backend for sklearn.

I think this could provide some nice performance boosts for dimensionality reduction and clustering.

jdrusso commented 1 year ago

Actually, the sklearn dimensionality reduction (PCA) and clustering (K-means) I use don't really use the joblib parallel backend, so this wouldn't really help anything anyways.