将xgboost.Booster的实例转换为实现scikit-learn API的模型

问题描述

我正在尝试使用mlflow保存模型,然后稍后加载以进行预测。

我正在使用xgboost.XGBRegressor模型及其sklearn函数.predict().predict_proba()进行预测,但事实证明mlflow不支持实现sklearn API,因此,以后从mlflow加载模型时,mlflow返回xgboost.Booster的实例,并且它不实现.predict().predict_proba()函数。

是否可以将xgboost.Booster转换回实现sklearn API函数的xgboost.sklearn.XGBRegressor对象?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)