带有GPU的GridSearch LightGBM

Question

如何使用GPU通过LightGBM进行GridSearch？如果您只想使用默认参数训练lgb模型，则可以执行以下操作：

dataset = lgb.Dataset(X_train, y_train)
lgb.train({'device': 'gpu'}, dataset)

要做GridSearch，最好做这样的事情：

lgbm_classifier = lgb.LGBMClassifier()

param_grid = {
    'learning_rate': [...],
    'n_estimators': [...],
}

GridSearchCV(lgbm_classifier, param_grid)

但是看起来lgb.LGBMClassifier()并不像device那样具有lgb.train()的参数。您如何设置它以使用GPU？

Answer 1

LGBMClassifier() 确实具有 device/device_type 参数，可以根据 parameter docs 将其设置为“gpu”。

要将其设置为“gpu”，您可以配置基于 GPU 的 LGBM。一些社区成员已经演示了如何做到这一点 here 和 here。第二个资源也有一个叫做 glances 的东西（在 PyPi.org 上）来观察 GPU 使用情况，解决了提出的意见。