请您参考如下方法:
该 tensorflow 示例已经具有计算相似度和显示最近单词的功能。 最简单的方法是使用该功能。
在第4步定义valid_examples后,你可以自己给词。
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
然后你可以在第一行看到结果。例如,我的结果是
Nearest to dog: empower, nephew, stationary, marmoset, wow, kvac, dasyprocta, centaur,
