使用tensorflow.js进行智能自动完成

时间:2019-05-04 06:24:48

标签: javascript tensorflow local-storage tensorflow.js

是否可以构建某种自动完成选择,以根据先前的输入来建议值?
假设我有5k项的机场选择。我想给用户提供前5个值,以根据他之前的偏好进行选择。

1 个答案:

答案 0 :(得分:0)

深度学习方法:

基本上,您想要的是RNN支持的字级预测模型。此类模型用于文本生成。您可以参考一个示例here

  

这些模型专门输入一个单词(t)并预测下一个单词(t1)。然后取t1即可预测t2。这个循环一直持续到达到最大单词数为止。

如果我必须在像

这样的单个句子上训练这样的RNN模型
Hello world from TensorFlow

然后功能标签对如下所示:

X, Y
Hello, World
World, from
from, TensorFlow

可以从用户的先前输入中训练这些句子。

构建此类模型所需的条件:

  1. 很多用户数据。
  2. 获得前N个预测的排名算法
  3. 可能是GPU或TPU。 CPU可能证明很慢,可能需要几天时间。

非深度学习方法:

您可以贝叶斯定理,也可以简单地检查在特定单词之后哪个单词具有最高的概率。在“早上”一词之后,“早上”一词的概率最高。

提示:

  • 通常,您还需要处理可能不会在字典中添加的自定义用户单词。它可能是一个人的名字。因此,在收集数据时,您需要将此类单词标记为一个常见单词,例如。