我在 GPU 机器上训练和验证了 XGBoost 模型并对其进行了pickle。当我加载泡菜并在同一台机器上进行测试时，它工作得很好。我得到了多班ROC_AUC和作品。当在另一台只有 CPU 空闲的机器上进行测试时，我加载 pickle，并根据需要设置：
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)
但现在所有的结果都属于我的多个班级中的第一个。我非常感谢这个问题的答案：我还遗漏了什么？
谨致问候，贾恩:)
附言！似乎没有保留这个问题，这是一个固定的问题。 https://github.com/h2oai/h2o4gpu/issues/625