machine-learning - 尽管已编译为 GPU 运行，但通过 Scikit learn API 的 XGB 似乎并未在 GPU 中运行

Question

虽然 XGB 被编译为在 GPU 上运行，但当通过 Scikit learn API 调用/执行时，它似乎并未在 GPU 上运行。

请告知这是否是预期行为

Answer 1

据我所知，Scikit learn API 目前不支持 GPU。您需要使用学习 API(例如 xgboost.train(...))。这还要求您首先将数据转换为 xgboost DMatrix。

示例:

params = {"updater":"grow_gpu"}
train = xgboost.DMatrix(x_train, label=y_train)
clf = xgboost.train(params, train, num_boost_round=10)

更新:

Scikit Learn API 现在通过 **kwargs 参数支持 GPU: http://xgboost.readthedocs.io/en/latest/python/python_api.html#id1