android - Tensorflow Lite 输入字符串

Question

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

标签： androidtensorflownlptensorflow-lite

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