我有一个Keras模型,然后将其转换为.tflite模型。
该模型是使用binary_crossentropy的二进制分类器。
使用keras,我可以调用model.predict()
以获得小数预测,例如0.96271
或者我可以调用model.predict_classes()
将其舍入到最接近的类,例如0或1。
tflite模型只是一个黑匣子,输入必须作为ByteBuffer提交,并且仅返回二进制输出(0或1)。
如何确保返回十进制值?我可以对Keras模型进行任何更改以确保自动返回小数吗?