我已经在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