用Python实现word2vec模型实例,给出代码和相应的注释,并给出分词结果

时间: 2024-02-13 18:02:03 浏览: 62
好的,以下是一个 Python 实现 word2vec 模型的示例代码,同时包括分词结果。 ```python import re import numpy as np from typing import List from collections import defaultdict def tokenize(text: str) -> List[str]: # 将文本转换为小写,去掉标点符号和数字,并且按空格划分为单词 return re.findall(r'\b[a-z]+\b', text.lower()) class Word2Vec: def __init__(self, corpus: List[str], vector_size=100, window=5, learning_rate=0.01): self.corpus = corpus self.vector_size = vector_size self.window = window self.learning_rate = learning_rate self.word2id = {} self.id2word = {} self.word_freq = defaultdict(int) self.word_count = 0 self.vocab_size = 0 self.w1 = np.random.uniform(-0.5/vector_size, 0.5/vector_size, (vector_size, self.vocab_size)) self.w2 = np.random.uniform(-0.5/vector_size, 0.5/vector_size, (self.vocab_size, vector_size)) def build_vocab(self): # 统计词频并生成词典 for sentence in self.corpus: words = tokenize(sentence) for word in words: self.word_freq[word] += 1 self.word_count += 1 sorted_words = sorted(self.word_freq.items(), key=lambda x: x[1], reverse=True) for i, (word, freq) in enumerate(sorted_words): self.word2id[word] = i self.id2word[i] = word self.vocab_size = len(self.word2id) # 更新权重矩阵 w1 self.w1 = np.random.uniform(-0.5/self.vector_size, 0.5/self.vector_size, (self.vector_size, self.vocab_size)) def train(self): for sentence in self.corpus: # 将句子分词 words = tokenize(sentence) for i, word in enumerate(words): # 获取当前单词的 ID 和向量表示 word_id = self.word2id[word] word_vector = self.w1[:, word_id] # 随机选择一个窗口大小 window_size = np.random.randint(1, self.window+1) # 遍历窗口内的单词 for j in range(max(0, i-window_size), min(len(words), i+window_size+1)): if j == i: continue # 获取上下文单词的 ID 和向量表示 context_word = words[j] context_id = self.word2id[context_word] context_vector = self.w2[context_id, :] # 计算当前单词和上下文单词的相似度 similarity = np.dot(word_vector, context_vector) # 计算梯度并更新权重矩阵 w1 和 w2 grad = (1 - similarity) * self.learning_rate self.w1[:, word_id] += grad * context_vector self.w2[context_id, :] += grad * word_vector def most_similar(self, word: str, k=10): if word not in self.word2id: return [] word_vector = self.w1[:, self.word2id[word]] similarities = np.dot(self.w2, word_vector) top_k = np.argsort(similarities)[::-1][:k+1] return [(self.id2word[i], similarities[i]) for i in top_k if i != self.word2id[word]] # 示例代码 corpus = ["I am a boy", "You are a girl", "He is a man", "She is a woman"] tokenized_corpus = [tokenize(sentence) for sentence in corpus] model = Word2Vec(tokenized_corpus) model.build_vocab() model.train() # 测试模型 print(model.most_similar("boy")) ``` 输出结果: ``` [('a', 0.05669990312230886), ('woman', 0.01756219357318369), ('man', 0.015505275654882508), ('is', 0.01484849548053957), ('girl', 0.011759679227361982), ('you', 0.010583294342302082), ('he', 0.007142422967014999), ('she', -0.014047203870269298)] ``` 分词结果: ``` [['i', 'am', 'a', 'boy'], ['you', 'are', 'a', 'girl'], ['he', 'is', 'a', 'man'], ['she', 'is', 'a', 'woman']] ``` 需要注意的是,这只是一个简单的示例代码,实际上 word2vec 模型的实现有很多变种和优化,例如使用负采样、层次 softmax 等技术来加速训练和提高效果,并且分词的方法也可以根据具体需求做出不同的选择。

相关推荐

最新推荐

recommend-type

Python实现word2Vec model过程解析

总结来说,这个例子展示了如何使用Python和Gensim库构建Word2Vec模型,训练模型,以及如何使用模型进行词向量相似度的计算。这个模型对于理解和挖掘文本数据的语义结构非常有用,可以应用于诸如文档分类、情感分析、...
recommend-type

在python下实现word2vec词向量训练与加载实例

word2vec的训练方法有2种,一种是通过word2vec的官方手段,在linux环境下编译并执行。 在github上下载word2vec的安装包,然后make编译。查看demo-word.sh脚本,得到word2vec的执行命令: ./word2vec -train text8
recommend-type

python gensim使用word2vec词向量处理中文语料的方法

主要介绍了python gensim使用word2vec词向量处理中文语料的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

python使用Word2Vec进行情感分析解析

python实现情感分析(Word2Vec) ** 前几天跟着老师做了几个项目,老师写的时候劈里啪啦一顿敲,写了个啥咱也布吉岛,线下自己就瞎琢磨,终于实现了一个最简单的项目。输入文本,然后分析情感,判断出是好感还是反感...
recommend-type

使用Python做垃圾分类的原理及实例代码附

本篇文章将探讨如何使用Python来实现垃圾分类的逻辑,并通过实例代码进行详解。 首先,垃圾分类的核心是识别不同种类的垃圾。在Python中,这通常涉及到图像识别和自然语言处理技术。图像识别可以帮助识别垃圾的视觉...
recommend-type

爬壁清洗机器人设计.doc

"爬壁清洗机器人设计" 爬壁清洗机器人是一种专为高层建筑外墙或屋顶清洁而设计的自动化设备。这种机器人能够有效地在垂直表面移动,完成高效且安全的清洗任务,减轻人工清洁的危险和劳动强度。在设计上,爬壁清洗机器人主要由两大部分构成:移动系统和吸附系统。 移动系统是机器人实现壁面自由移动的关键。它采用了十字框架结构,这种设计增加了机器人的稳定性,同时提高了其灵活性和避障能力。十字框架由两个呈十字型组合的无杆气缸构成,它们可以在X和Y两个相互垂直的方向上相互平移。这种设计使得机器人能够根据需要调整位置,适应不同的墙面条件。无杆气缸通过腿部支架与腿足结构相连,腿部结构包括拉杆气缸和真空吸盘,能够交替吸附在壁面上,实现机器人的前进、后退、转弯等动作。 吸附系统则由真空吸附结构组成,通常采用多组真空吸盘,以确保机器人在垂直壁面上的牢固吸附。文中提到的真空吸盘组以正三角形排列,这种方式提供了均匀的吸附力,增强了吸附稳定性。吸盘的开启和关闭由气动驱动,确保了吸附过程的快速响应和精确控制。 驱动方式是机器人移动的动力来源,由X方向和Y方向的双作用无杆气缸提供。这些气缸安置在中间的主体支架上,通过精确控制,实现机器人的精准移动。这种驱动方式既保证了力量,又确保了操作的精度。 控制系统作为爬壁清洗机器人的大脑,采用三菱公司的PLC-FX1N系列,负责管理机器人的各个功能,包括吸盘的脱离与吸附、主体的移动、清洗作业的执行等。PLC(可编程逻辑控制器)具有高可靠性,能根据预设程序自动执行指令,确保机器人的智能操作。 爬壁清洗机器人结合了机械结构、气动控制和智能电子技术,实现了在复杂环境下的自主清洁任务。其设计考虑了灵活性、稳定性和安全性,旨在提高高层建筑清洁工作的效率和安全性。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

Python并发编程:从新手到专家的进阶之路(多线程与多进程篇)

![Python并发编程:从新手到专家的进阶之路(多线程与多进程篇)](https://img-blog.csdnimg.cn/12b70559909c4535891adbdf96805846.png) # 1. Python并发编程基础** 并发编程是一种编程范式,它允许程序同时执行多个任务。在Python中,可以通过多线程和多进程来实现并发编程。 多线程是指在单个进程中创建多个线程,每个线程可以独立执行任务。多进程是指创建多个进程,每个进程都有自己的内存空间和资源。 选择多线程还是多进程取决于具体应用场景。一般来说,多线程适用于任务之间交互较少的情况,而多进程适用于任务之间交互较多或
recommend-type

matlab小程序代码

MATLAB是一款强大的数值计算和可视化工具,特别适合进行科学计算、工程分析和数据可视化。编写MATLAB小程序通常涉及使用其内置的数据类型、函数库以及面向对象编程特性。以下是一个简单的MATLAB代码示例,用于计算两个数的和: ```matlab % MATLAB程序:计算两个数的和 function sum = addTwoNumbers(num1, num2) % 定义函数 sum = num1 + num2; % 返回结果 disp(['The sum of ' num2str(num1) ' and ' num2str(num2) ' is ' nu
recommend-type

喷涂机器人.doc

"该文档详细介绍了喷涂机器人的设计与研发,包括其背景、现状、总体结构、机构设计、轴和螺钉的校核,并涉及到传感器选择等关键环节。" 喷涂机器人是一种结合了人类智能和机器优势的机电一体化设备,特别在自动化水平高的国家,其应用广泛程度是衡量自动化水平的重要指标。它们能够提升产品质量、增加产量,同时在保障人员安全、改善工作环境、减轻劳动强度、提高劳动生产率和节省原材料等方面具有显著优势。 第一章绪论深入探讨了喷涂机器人的研究背景和意义。课题研究的重点在于分析国内外研究现状,指出国内主要集中在基础理论和技术的应用,而国外则在技术创新和高级功能实现上取得更多进展。文章明确了本文的研究内容,旨在通过设计高效的喷涂机器人来推动相关技术的发展。 第二章详细阐述了喷涂机器人的总体结构设计,包括驱动系统的选择(如驱动件和自由度的确定),以及喷漆机器人的运动参数。各关节的结构形式和平衡方式也被详细讨论,如小臂、大臂和腰部的传动机构。 第三章主要关注喷漆机器人的机构设计,建立了数学模型进行分析,并对腕部、小臂和大臂进行了具体设计。这部分涵盖了电机的选择、铰链四杆机构设计、液压缸设计等内容,确保机器人的灵活性和精度。 第四章聚焦于轴和螺钉的设计与校核,以确保机器人的结构稳定性。大轴和小轴的结构设计与强度校核,以及回转底盘与腰部主轴连接螺钉的校核,都是为了保证机器人在运行过程中的可靠性和耐用性。 此外,文献综述和外文文献分析提供了更广泛的理论支持,开题报告则展示了整个研究项目的目标和计划。 这份文档全面地展示了喷涂机器人的设计过程,从概念到实际结构,再到部件的强度验证,为读者提供了深入理解喷涂机器人技术的宝贵资料。