首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >将xgboost.Booster类转换为XGBRegressor或从xgboost.Booster加载XGBRegressor

将xgboost.Booster类转换为XGBRegressor或从xgboost.Booster加载XGBRegressor
EN

Stack Overflow用户
提问于 2022-05-03 09:22:42
回答 1查看 268关注 0票数 0

我从萨格梅克那里得到了一个型号:

代码语言:javascript
运行
复制
<class 'xgboost.core.Booster'>

我可以在本地得分,这是很棒的,但一些谷歌搜索已经表明,它可能不可能做这样的“标准”的事情从这里取得。

代码语言:javascript
运行
复制
plt.barh(boston.feature_names, xgb.feature_importances_)

是否可以将xgboost.core.Booster转换为XGBRegressor?也许人们可以使用save_raw方法查看?谢谢!

到目前为止我试过:

代码语言:javascript
运行
复制
xgb_reg = xgb.XGBRegressor() 
xgb_reg._Boster = model
xgb_reg.feature_importances_

但结果是:

代码语言:javascript
运行
复制
NotFittedError: need to call fit or load_model beforehand
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-05-03 14:32:47

沿着这些思路的东西似乎运转良好:

代码语言:javascript
运行
复制
local_model_path = "model.tar.gz"
with tarfile.open(local_model_path) as tar:
    tar.extractall()

model = xgb.XGBRegressor() 
model.load_model(model_file_name)     

然后,模型就可以像往常一样使用-- model.tar.gz是一种来自sagemaker的人造工具。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/72097296

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档