run GeneticSelectionCV on gpu python

user_10666991 0
Viewed 89

I train GeneticSelectionCV model on cpu with the following code:

from genetic_selection import GeneticSelectionCV
from sklearn.naive_bayes import GaussianNB
g = GeneticSelectionCV(estimator=GaussianNB(), scoring='accuracy').fit(X, y)

How can I train GaussianNB and GeneticSelectionCV on GPU?

scikit-learn genetic-algorithm gpu python
1 Answers

I'm not aware of whether this has been tested, but you can likely use cuML's GPU-accelerated GaussianNB with GeneticSelectionCV as a drop-in replacement for scikit-learn's GaussianNB. This recent blog provides some information about potential performance.

user_60665280
Related