У меня есть модель Keras, которую я затем преобразовал в модель .tflite.
Модель представляет собой двоичный классификатор, использующий binary_crossentropy.
С помощью keras я могу позвонить model.predict(), чтобы получить десятичное предсказание, например, 0,96271
, или я могу позвонить model.predict_classes(), который округляет его до ближайшего класса, например, 0 или 1.
Модель tflite - это просто черный ящик, входные данные должны быть представлены как ByteBuffer, и возвращается только двоичный выход (0 или 1).
Как я могу гарантировать, что возвращается десятичное значение?Могу ли я внести какие-либо изменения в модель Keras, чтобы автоматически возвращать десятичное число?