tensorflow 如何从这个输出中只返回一个元素?[duplicate]

mefy6pfw  于 2023-02-13  发布在  其他
关注(0)|答案(1)|浏览(101)
    • 此问题在此处已有答案**:

Convert list or numpy array of single element to float in python(5个答案)
2天前关闭。
我在Python中使用TensorFlow。使用我的模型进行预测时,我的输入如下:

UberLyft_model.predict(my_UberLyft_encoded)

我得到这个输出:

1/1 [==============================] - 0s 53ms/step
array([[16.11945]], dtype=float32)

但我只想要这个输出:
16.11945

bq3bfh9z

bq3bfh9z1#

试试看

prediction = UberLyft_model.predict(my_UberLyft_encoded)
result = prediction[0][0]
print(result)

相关问题