python - 我用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 是一种简单的算法，我认为它可以很容易地用任何编程语言实现。我建议你用母语创建一个算法。您将对它的工作有更多的控制流..