如何在张量流

Question

如何给出一个特定的单词来测试训练有素的word2vec模型？

例如

输入“狗”，并返回像“猫”，“鸟”等的近词。

谢谢！

word2vec_basic_py

Answer 1

张量流示例已经具有计算相似性并显示最近单词的功能。 最简单的方法是使用该功能。

在定义valid_examples后的第4步中，您可以提供自己的单词。

valid_examples = np.random.choice(valid_window, valid_size, replace=False)
num_sampled = 64    # Number of negative examples to sample.

sample_word = "dog";
if sample_word in dictionary:
  sample_index = dictionary[sample_word]
else:
  sample_index = 0  # dictionary['UNK']
valid_examples[0] = sample_index

然后你可以在第一行看到结果。例如，我的结果是

  

最近的狗：授权，侄子，静止，mar猴，哇，kvac，   dasyprocta，半人马，