使sagemaker端点返回概率

时间:2020-09-03 01:13:21

标签: amazon-sagemaker

我可以调用sagemaker端点并返回预测:

predictor = sagemaker.predictor.RealTimePredictor(endpoint="sleep", 
    sagemaker_session=session)
predictor.content_type="text/csv"
predictor.serializer=csv_serializer
predictor.deserializer=None
predictor.predict(data[0])
    
b'False\n'

我需要它返回一个概率,该概率指示其预测的强度,而不仅仅是其预测的标签。

如何使Sagemaker端点不仅返回预测的标签,还返回概率?

0 个答案:

没有答案