如何使用sagemaker下载的模型xgb booster对象绘制部分依赖关系图？

Question

我有一个XGBoost模型，在Sagemaker接受过培训，并在本地网站上下载。解压tar.gz文件时，会得到一个模型对象（booster）文件。 我想使用Sagemaker模型实例为我的数据集的某些功能（使用预测进行批量转换）绘制部分依赖关系图。但是，我得到了以下错误

ValueError:“估计器”必须是拟合回归器或分类器。

我用来绘制图表的代码如下-

from sklearn.inspection import partial_dependence
pardep = partial_dependence(model, X, 'num_org')

这里的模型是<；0x7f9b7b9a2250处的xgboost.core.Booster>； X是一个批处理转换的数据框，有大约100列，“num_org”是我想要绘制的关于预测的图