移动端AI与ML应用开发：基于iOS和Android

208

|

第

11

章

让我们看看构建这样的应用程序需要什么！我假设你已经创建了一个应用程序，添加

了前面所示的 TensorFlow Lite pod，并创建了一个故事板，其中包含一个用于输入的

UITextView（带有一个名为

txtInput

的输出）、一个用于输出的 UILabel（带有一个名为

txtOutput

的输出），以及一个名为

classifySentence

的按钮上的动作。完整的应用程

序在本书的存储库中，我会给你介绍需要执行的 NLP 特定的编码。

11.4.1

第

1

步：加载词汇

当你使用 Model Maker 创建模型时（请参阅第 8 章），可以从 Colab 环境下载模型和词汇

文件。vocab 文件只是被称为没有扩展名的

vocab

，因此将其重命名为

vocab.txt

并将其

添加到你的应用程序中。确保它包含在应用程序包中，否则你的应用程序将无法在运行

时读取它。

要使用词汇，你需要一个字典，它是一组键值对。键是一个字符串（包含单词），值是一

个

int

（包含单词的索引），如下所示：

var words_dictionary = [String : Int]()

要加载字典，你可以编写一个名为

loadVocab()

的辅助函数。让我们来探索一下它的作

用。首先，通过定义

filePath

将

vocab.txt

指定为要加载的文件：

if let filePath = Bundle.main.path( forResource: "vocab",

ofType: "txt") ...