如何加载使用（版本 = '最新'）框架训练的 Sagemaker XGBoost 模型？

Question

管道中有一个使用此容器创建的现有 xgboost 模型

sagemaker.image_uris.retrieve('xgboost', sagemaker.Session().boto_region_name, version='latest')

输出：

'{accountid}.dkr.ecr.us-east-1.amazonaws.com/xgboost:latest'

我从模型工件中提取了 model.tar.gz 并加载了 xgboost-model 文件
但它给出了这个错误

XGBoostError: basic_string::resize

我运行了一个 shell 脚本，使用所有可用的 XGBoost 版本加载模型，但没有任何效果。
我只是想检查功能重要性

using model.get_score

。

Answer 1

作为调查此错误的第一步，请通过在本地安装 xgboost 来检查此错误是否在您的本地环境中重现。

如果此错误也在本地重现，您可以从本文中删除 amazon-web-services / amazon-sagemaker 标签。

还建议添加一些信息，了解您如何调用 xgboost API 来加载模型，以及您的模型文件在提取的模型目录下的结构，因为错误可能是由于模型的结构以及如何构建而发生的。加载它。