稍后在CPU机器上运行GPU训练的XGBoost模型

时间:2019-01-13 13:44:32

标签: gpu cpu xgboost

我已经在GPU机器上训练并验证了XGBoost模型并将其腌制。当我在同一台机器上加载泡菜并进行测试时,它可以完美工作。我得到了多类ROC_AUC和作品。在另一台只有CPU备用的机器上进行测试时,我会加载泡菜,并根据需要设置:

xgboost_clf = xgboost_clf.set_params(predictor = 'cpu_predict')
xgboost_clf = xgboost_clf.set_params(tree_method = 'hist')

我运行它:

Y_test_pred_xgboost = xgboost_clf.predict(X_test)

但是现在所有结果都属于我的多个类中的第一个。我将不胜感激地回答这个问题:我还有什么遗漏?

诚挚的问候,   简:)

PS!似乎没有保留这个问题,这是一个固定的问题。 https://github.com/h2oai/h2o4gpu/issues/625

0 个答案:

没有答案