
在
Android
中使用自定义模型
|
189
10.4
使用语言模型
在构建使用语言的模型时,该模式与你在图 10-1 中看到的非常相似,具体情况如图 10-7
所示。
一个主要区别是,你的应用程序使用基于自然语言处理(NLP)的模型,需要与训练底
层模型相同的单词词典。回忆第 9 章,句子被分成单词列表,单词被赋予数字标记。为
这些标记学习向量,为该词建立情绪。例如,单词“ dog”可能被赋予标记 4,并且可
以为标记 4 学习像 [0, 1, 0, 1] 这样的多维向量。然后在你的应用程序中可以使用字典来
映射单词“ dog”到 4。该模型还在固定长度的句子上进行训练,你的应用程序还需要
知道该数据。
包装代码
字典 元数据
模型
张量 张量
Android 数据
(
例如
,
字符串
)
Android 数据
(
例如
,
字符串
)
图
10-7
:在
NLP
应用程序中使用模型
如果你使用 TensorFlow Lite Model Maker 构建模型,那么元数据和字典实际上会编译到
.tflite
文件中,让你的生活更轻松。
对于本节的其余部分,我假设你有一个 NLP 模型,使用 Model Maker 进行训练,就像
第 9 章中演示的情绪分类器一样。你也可以在本章的代码库中找到已为你实现的,其中
完整的应用程序,包括模型。
10.5
创建用于语言分类的
Android
应用程序
使用 Android Studio 创建一个新的 Android 应用程序。只需让它成为一个带有空活动的
简单活动。完成后,编辑 build.gradle ...