我正在尝试使用mlflow来保存模型,然后加载它以进行预测。我使用了一个xgboost.XGBRegressor模型和它的sklearn函数.predict()和.predict_proba()来进行预测,但是事实证明mlflow并不支持实现sklearn API的模型,所以当稍后从mlflow加载模型时,mlflow返回一个xgboost.Booster的实例,并且它没有实现.predict()或.predict_proba()函数。有没有办法将xgboost.Booster转换回实现sklearn A
‘'bst’是我在R中构建的xgboost模型的名称,它给出了使用这段代码的测试数据集的预测值。因此,它肯定是一个xgboost模型。xdtest) # get prediction in test sample现在,我想保存模型,以便另一个可以使用模型与他们的数据集,其中有相同的预测变量和同样的变量要预测与xgboost.pdf的第4页( xgboost包的文档)一致,我使用了xgb.save命令:
xg