前言
炎炎夏日马上就要到了，可是疫情肆虐的两年里，我已经两年没有出去旅游了，淡忘了许多关于“沙滩”和“美女”的美景，作为一个技术人员，我想通过程序找出那些逝去的记忆，“沙滩”和“美女”以及其他相关美好的回忆将再次激荡在我脑中！
思路
学习过 NLP 的童鞋肯定知道，想找出某个词最相关的词，简单粗暴的方法就是使用 word2vec ，语义的可解释性也比较好，所以就定下用它！
先使用中文训练好一个 word2vec 模型，然后再通过关键词语的词向量互相组合，不断扩展出和最初关键词相近的词语，这样我们就能组成一个大的词语网络，这个词语网络就能将美好的场景重新描绘在我们脑海中。是不是很简单？
准备
需要懂得 word2vec 的基本原理
需要掌握基础的 python 和简单算法基础
需要安装 gensim 库，学会使用基本的 word2vec 的相关操作
听懂掌声???
实现
import collectionsimport itertoolsfrom gensim.models import KeyedVectorsmodel = KeyedVectors.load_word2vec_format('./word2vec_779845.bin', binary=True)words = ['沙滩', '美女', '比基尼']N = 1visited = set(words)result = collections.defaultdict(list)for L in range(2, 4): for word in itertools.combinations(words, L): if word not in visited: visited.add(word) vectors = model.most_similar(list(word), topn=5) for i, item in enumerate(vectors): if item[0] not in visited: words.append(item[0]) visited.add(item[0]) result[word].append(item[0])print(words)print(result)结果分析
经过一把程序梭哈我们可以得出的词语如下所示，看了这些词语，我已经有了心潮澎湃了，兄弟们有没有心动呢？word2vec 真的是我的良师益友，它已经帮我找出了“女郎”、“比基尼”、“泳装”，也帮我找出了“派对”、“草地”、“音乐会”，我放佛又回到了那个激情四射的夜晚沙滩派对，俊男靓女在参加音乐会，大家都在唱歌、跳舞、玩耍，尽情享受其中，活活美死！！
['沙滩', '美女', '比基尼', '海滩', '摇滚', '女郎', '闪耀', '泳装', '红毯', '群星', '闪亮', '性感', '主角', '女王', '鸟巢', '派对', '当红', '偶像', '英伦', '璀璨', '帅气', '装扮', '女主角', '高手', '身着', '盛装', '热辣', '古装', '一袭', '音乐会', '舞', '驰骋', '风情', '风帆', '城堡', '巡游', '熠熠', '草地', '景色', '复古', '裙装', '迷人', '化身', '压轴', '火辣', '帅哥', '男星', '耀眼', '足球宝贝', '女星', '大腕', '浪漫', '刘亦菲', '男主角', '情侣', '巨星', '惊艳', '倾情', '华丽', '礼服', '妩媚', '音乐节', '首映礼', '妖娆', '唯美', '硬汉', '靓丽']
另外我们可以使用多个词语描绘的场景来找出相关的词语描述，部分结果如下所示。我们可以看到“沙滩”和“美女”相关的词有“海滩”、“摇滚”、“女郎”、“闪耀”，此时我已经沉浸其中在沙滩上的摇滚现场群星闪耀，美女们在尽情 ROCK ，不要叫醒我，让我多睡一会 ...
{('沙滩', '美女'): ['海滩', '摇滚', '女郎', '闪耀'], ('沙滩', '比基尼'): ['泳装', '红毯', '群星', '闪亮'], ('美女', '比基尼'): ['性感', '主角'], ('沙滩', '美女', '比基尼'): ['女王'], ('沙滩', '美女', '海滩'): ['鸟巢', '派对'], ('沙滩', '美女', '摇滚'): ['当红', '偶像'], ('沙滩', '美女', '闪耀'): ['英伦', '璀璨'], ('沙滩', '美女', '性感'): ['帅气', '装扮'], ('沙滩', '美女', '主角'): ['女主角', '高手'], ('沙滩', '比基尼', '海滩'): ['身着', '盛装'], ('沙滩', '比基尼', '摇滚'): ['热辣', '古装'], ('沙滩', '比基尼', '泳装'): ['一袭'], ('沙滩', '比基尼', '群星'): ['音乐会'], ('沙滩', '海滩', '摇滚'): ['舞', '驰骋', '风情'], ('沙滩', '海滩', '女郎'): ['风帆'], ('沙滩', '海滩', '泳装'): ['城堡', '巡游'], ('沙滩', '海滩', '红毯'): ['熠熠', '草地'], ('沙滩', '海滩', '性感'): ['景色'], ('沙滩', '摇滚', '性感'): ['复古'], ('沙滩', '女郎', '性感'): ['裙装'], ('沙滩', '闪耀', '性感'): ['迷人'], ('沙滩', '闪耀', '主角'): ['化身'], ('沙滩', '红毯', '群星'): ['压轴'], ('美女', '比基尼', '性感'): ['火辣'], ('美女', '比基尼', '主角'): ['帅哥'], ('美女', '海滩', '摇滚'): ['男星'], ('美女', '海滩', '闪耀'): ['耀眼'], ('美女', '海滩', '泳装'): ['足球宝贝'], ('美女', '海滩', '主角'): ['女星'], ('美女', '摇滚', '群星'): ['大腕'], ('美女', '摇滚', '性感'): ['浪漫'], ('美女', '摇滚', '主角'): ['刘亦菲', '男主角'], ('美女', '女郎', '主角'): ['情侣'], ('美女', '红毯', '主角'): ['巨星'], ('美女', '闪亮', '性感'): ['惊艳'], ('比基尼', '摇滚', '群星'): ['倾情'], ('比基尼', '摇滚', '性感'): ['华丽'], ('比基尼', '泳装', '性感'): ['礼服'], ('海滩', '摇滚', '性感'): ['妩媚'], ('海滩', '红毯', '群星'): ['音乐节', '首映礼'], ('摇滚', '女郎', '性感'): ['妖娆'], ('摇滚', '闪耀', '闪亮'): ['唯美'], ('闪耀', '泳装', '群星'): ['硬汉'], ('闪耀', '红毯', '性感'): ['靓丽']}尾记
本文情节纯属虚构，请勿乱想，主要是使用 word2vec 来实现一些好玩的东西，然后去掘金薅羊毛? ? ? 。
原文：https://juejin.cn/post/7101199425489338376