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()参数。你如何设置它以使用GPU？devicelgb.train()

Answer 1

LGBMClassifier() 确实有device/device_type参数，可以根据参数 docs设置为 'gpu' 。

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