标签： amazon-sagemaker
我使用Sagemaker框架训练了一个Xgboost多级分类算法，用于5个类。然后我在S3上保存了模型。现在，当我加载它时，我只能选择预测，这将返回类，而不是每个类的概率。
pred = xgb_uploaded_model.predict(new_data)
上传的模型没有predict_proba的选项，这本来就派上用场了。 关于从这个模型中获取概率的热点的想法是这样保存的吗？ 附：我不想使用端点。