如何从keras/.tfite模型得到非二进制输出？

2024-04-29 09:56:22 发布

您现在位置：Python中文网/ 问答频道 /正文
1513 3

我有一个Keras模型，然后我将其转换为.tflite模型。你知道吗

该模型是一个基于二元交叉熵的二元分类器。你知道吗

使用keras，我可以调用model.predict()来获得十进制预测，例如0.96271

或者我可以调用model.predict_classes()，将其舍入到最近的类，如0或1。你知道吗

tflite模型只是一个黑盒，输入必须作为ByteBuffer提交，并且只返回二进制输出（0或1）。你知道吗

如何确保返回十进制值？我是否可以对Keras模型进行任何更改，以确保自动返回小数点？你知道吗


Tags： 模型model分类器二进制黑盒predict交叉classes
0条回答

目前没有回答

相关问题 更多 >

    编程相关推荐