python - 带 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 使用情况，它解决了所提出的评论。