知道如何在 tensorflow Lite 中使用語言 model 嗎？

Question

有誰知道如何在 TensorFlow Lite 中使用語言 model ？ 我有一個生成的語言 model，在 TensorFlow 中有一個LSTM結構。 我已經將它轉換為.tflite用於 android。 現在我的問題是如何使用它？ 我的意圖是使用 model 來預測句子中的下一個單詞。 The original model works perfectly in Python, now what I need is for it to work and make predictions in Java or Kotlin.

Answer 1

您可以通過兩種方式實現此目的：

1. 在assets文件夾中添加.tflite model，實現構建Gradle文件，最后導入model實現java/kotlin中的需求。 如此處所示
2. 通過使用 firebase、firebase 提供自定義機器學習 model 部署選項。 您可以將 model 添加到 firebase 並將其用作遠程服務。 文檔的鏈接。

如果您的 model 尺寸很小並且您想離線運行您的應用程序，我會推薦第一種方法。