I built a random forest classifier using the Sklearn Python API. After hearing about the h2o4gpu package for sklearn GPU acceleration I installed it and imported it. But the ensemble.RandomForestClassifier still seems to be running on the CPU. Am I missing something??
Asked
Active
Viewed 188 times
1 Answers
0
Based on specified parameters Sklearn can be used as a fallback. Also you can force to use h2o's back end by
backend='h2o4gpu'
sh1ng
- 2,808
- 4
- 24
- 38