h2o4gpu не запускает ensemble.RandomForestCLassifier на графическом процессоре - PullRequest
0 голосов
/ 21 мая 2019

Я построил классификатор случайных лесов, используя Sklearn Python API. Услышав о пакете h2o4gpu для ускорения графического процессора sklearn, я установил его и импортировал. Но ensemble.RandomForestClassifier все еще работает на процессоре. Я что-то упустил ??

1 Ответ

0 голосов
/ 21 мая 2019

На основании указанных параметров Sklearn может использоваться как запасной вариант. Также вы можете заставить использовать h2o back end backend='h2o4gpu'

...