word2vec中Skip-Gram和CBOW

本文详细介绍了Word2Vec模型中的两种主要方法:CBOW和Skip-Gram。CBOW通过上下文预测中间词,而Skip-Gram则以中心词预测上下文。文中还给出了训练神经网络的示例,并探讨了两种模型的训练参数及其作用。
摘要由CSDN通过智能技术生成

Word2Vec是从大量文本语料中以无监督的方式学习语义知识的一种模型,它被大量地用在自然语言处理(NLP)中。Word2Vec模型中,主要有Skip-Gram和CBOW两种模型,本文将详细介绍这两种模型。

CBOW

CBOW全称是Continuous Bag of Words Model,即连续词袋模型,模型如下图(图为七月在线NLP课程截图)
在这里插入图片描述
由图可看出,CBOW的输入是上下文,输出是中间词,即是通过上下文内容来预测中间词。

接下来我们来看看如何训练我们的神经网络。假如我们有一个句子“The dog barked at the mailman”。
定义两个参数batch_size和bag_window,batch_size定义了batch个数,bag_window定义了上下文长度,假设我们设置batch_size为4,bag_window为1,那么我们得到的batch为:[['The', 'barked'], ['dog', 'at'], ['barked', 'the'], ['at,'mailman']],对应的labels为['dog', 'barked', 'at', 'the']将batch作为x,label作为y喂入构造的网络中就可以生成一个概率分布

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值