xgboost:使用GPU进行训练,使用CPU进行评估

时间:2019-08-05 14:51:22

标签: python xgboost

我已经使用GPU训练了XGBoost模型。如果我加载它,它会显示

xgb_mod.attributes()
{'SAVED_PARAM_gpu_id': '2', 'SAVED_PARAM_predictor': 'gpu_predictor'}

最初,我尝试将其加载到没有GPU的笔记本电脑上,并在新的样本上运行它,但是所有的预测结果都是荒谬的。如果我在训练它的服务器上运行它,则所有预测都可以。

是否可以在非GPU设置中使用该模型?

0 个答案:

没有答案