Tensorflow Lite 输入字符串

Question

我想在我的 android 应用上运行一个 TfLite 模型来进行一些自然语言处理。我想知道我必须以什么格式输入 TFLite 模型？我找不到必要的文件。

Answer 1

这是一个使用 TensorFlow Lite 训练文本分类模型的示例。 https://www.tensorflow.org/lite/tutorials/model_maker_text_classification

并查看此示例应用，了解如何将模型集成到 Android 应用。 https://github.com/tensorflow/examples/tree/master/lite/examples/text_classification/android

TF Lite 目前不支持原始文本输入。您需要将文本输入预处理为数字（例如词嵌入、标记化等），然后将它们用作模型输入。