6

我已经使用 Sagemaker 框架为 5 个类训练了一个 Xgboost 多类分类算法。然后我将模型保存在 S3 上。现在,当我加载它时,我只有预测选项,这将返回我的类而不是每个类的概率。

pred = xgb_uploaded_model.predict(new_data)

上传的模型没有 predict_proba 选项,这会派上用场。有什么想法可以从这个保存的模型中获得概率吗?PS我不想使用端点。

4

0 回答 0