我已经在python中完成了k-nn模型，如何将其保存到tflite？

Question

有一种方法可以将k附近的邻居模型转换为tflite？据我所知'.pb不保存数据集值，仅存储图形和变量

Answer 1

TensorFlow Lite模型不会保存数据集值。这是保存TensorFlow图的更干净的方法。

对于KNN，您需要在TF中创建此算法，然后使用tf.lite.TFLiteConverter.from_session_graph()指定KNN的输入和输出张量。

converter = tf.lite.TFLiteConverter.from_session_graph( inputs, outputs )
buffer = converter.convert()
open( "model.tflite", 'wb' ).write(buffer)

inputs可能是模型的占位符张量，outputs可能是为预测类提供类标签的张量。

提示：

KNN是一种简单的算法，我认为可以用任何编程语言轻松实现。我建议您使用本地语言创建一个算法。您将对其进行更多控制。