0

我使用 Sklearn Python API 构建了一个随机森林分类器。在听说了用于 sklearn GPU 加速的 h2o4gpu 包后，我安装并导入了它。但ensemble.RandomForestClassifier似乎仍然在 CPU 上运行。我错过了什么吗？

4

1 回答 1

0

基于指定的参数，Sklearn 可以用作后备。您也可以通过以下方式强制使用 h2o 的后端 backend='h2o4gpu'

于 2019-05-21T16:55:17.690 回答