有人知道如何在tensorflow lite中使用语言模型吗？我有一个生成的语言模型 LSTM tensorflow 中的结构。我已经把它改成了 .tflite 在android上使用。现在我的问题是如何使用它？我的意图是用这个模型来预测句子中的下一个单词。最初的模型在python中工作得非常好，现在我需要的是让它在java或kotlin中工作并进行预测。
LSTM
.tflite
41ik7eoe1#
您可以通过两种方式实现这一点：添加 .tflite 在assets文件夹中建模，实现它以构建gradle文件，然后最终导入该模型以在java/kotlin中执行需求。如图所示通过使用firebase，firebase提供了定制的机器学习模型部署选项。您可以将模型添加到firebase并将其用作远程服务。指向文档的链接。我建议第一种方法，如果你的模型尺寸很小&如果你想离线运行你的应用程序。
1条答案
按热度按时间41ik7eoe1#
您可以通过两种方式实现这一点：
添加
.tflite在assets文件夹中建模，实现它以构建gradle文件，然后最终导入该模型以在java/kotlin中执行需求。如图所示通过使用firebase，firebase提供了定制的机器学习模型部署选项。您可以将模型添加到firebase并将其用作远程服务。指向文档的链接。
我建议第一种方法，如果你的模型尺寸很小&如果你想离线运行你的应用程序。