我正在对tensorflow服务模型运行预测,我将这个PredictResponse对象作为输出返回: 结果: outputs { key: "outputs" value { dtype: DT_FLOAT tensor_shape { dim { size: 1 } dim { size: 20 } } float_val: 0.000343723397236
结果:
outputs { key: "outputs" value { dtype: DT_FLOAT tensor_shape { dim { size: 1 } dim { size: 20 } } float_val: 0.000343723397236 float_val: 0.999655127525 float_val: 3.96821117632e-11 float_val: 1.20521548297e-09 float_val: 2.09611101809e-08 float_val: 1.46216549979e-09 float_val: 3.87274603497e-08 float_val: 1.83520256769e-08 float_val: 1.47733780764e-08 float_val: 8.00914179422e-08 float_val: 2.29388191997e-07 float_val: 6.27798826258e-08 float_val: 1.08802950649e-07 float_val: 4.39628813353e-08 float_val: 7.87182985462e-10 float_val: 1.31638898893e-07 float_val: 1.42612295306e-08 float_val: 3.0768305237e-07 float_val: 1.12661648899e-08 float_val: 1.68554503688e-08 } }
我想把浮动值作为列表.或者,或者,返回argmax float_val的值/索引!
这是由以下生成的:
stub = prediction_service_pb2.beta_create_PredictionService_stub(channel)
result = stub.Predict(request, 200.0)
感谢您的帮助.
答案是:floats = result.outputs['outputs'].float_val