如何从keras / .tfite模型获取非二进制输出?

时间:2019-03-22 00:38:57

标签: python tensorflow keras

我有一个Keras模型,然后将其转换为.tflite模型。

该模型是使用binary_crossentropy的二进制分类器。

使用keras,我可以调用model.predict()以获得小数预测,例如0.96271

或者我可以调用model.predict_classes()将其舍入到最接近的类,例如0或1。

tflite模型只是一个黑匣子,输入必须作为ByteBuffer提交,并且仅返回二进制输出(0或1)。

如何确保返回十进制值?我可以对Keras模型进行任何更改以确保自动返回小数吗?

0 个答案:

没有答案