我可以调用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端点不仅返回预测的标签,还返回概率?