10分钟快速入门 PyTorch (8) – Word Prediction

PyTorch入门实战教程

前面一篇文章中,我们简单的介绍了自然语言处理中最简单的词向量 word embedding,这一篇文章我们将介绍如何使用word embedding做自然语言处理的词语预测。

N-Gram language Modeling

首先我们介绍一下 N-Gram 模型。在一篇文章中,每一句话有很多单词组成,而对于一句话,这些单词的组成顺序也是很重要的,我们想要知道在一篇文章中我们是否可以给出几个词然后预测这些词后面的一个单词,比如’I lived in France for 10 years, I can speak _ .’那么我们想要做的就是预测最后这个词是French。

知道了我们想要做的事情之后,我们就可以引出 N-Gram 模型了。先给出其公式

这就是一个条件概率,也就是我们给定想要预测的单词的前面几个单词,然后最大化我们想要预测的这个单词的概率。

Code

数据预处理

首先我们给出了一段文章作为我们的训练集

CONTEXT_SIZE表示我们想由前面的几个单词来预测这个单词,这里设置为2,就是说我们希望通过这个单词的前两个单词来预测这一个单词。 EMBEDDING_DIM表示word embedding的维数,上一篇已经介绍过了。

接下来我们需要将数据整理好,也就是我们需要将单词三个分组,每个组前两个作为传入的数据,而最后一个作为预测的结果

接下来需要给每个单词编码,也就是用数字来表示每个单词,这样才能够传入word embeding得到词向量。

定义模型

这个模型需要传入的参数是所有的单词数,预测单词需要的前面单词数,即CONTEXT_SIZE,词向量的维度。

然后在向前传播中,首先传入单词得到词向量,比如在该模型中传入两个词,得到的词向量是(2, 100),然后将词向量展开成(1, 200),然后传入一个线性模型,经过relu激活函数再传入一个线性模型,输出的维数是单词总数,可以看成一个分类问题,要最大化预测单词的概率,最后经过一个log softmax激活函数。

然后定义好模型、loss以及优化函数。

训练

接着进行训练,一共跑100个epoch,在每个epoch中,word代表着预测单词的前面两个词,label表示要预测的词,然后记住需要将他们转换成Variable,接着进入网络得到结果,然后通过loss函数得到loss进行反向传播,更新参数。

训练完100个epoch后的结果如下

我们发现loss已经降到了0.37,也可以通过预测来检测我们的模型是否有效

得到的如下结果

可以发现我们能够准确地预测这个单词。

以上我们介绍了如何通过最简单的单边 N-Gram 模型预测单词,还有一种复杂一点的N-Gram模型通过双边的单词来预测中间的单词,这种模型有个专门的名字,叫 Continuous Bag-of-Words model (CBOW),具体的内容差别不大,就不再细讲了,代码的实现放在了github上面。

下一章我们将讲一下如何使用LSTM来做自然语言处理。

本文代码已经上传到了github上。

文章来源:知乎专栏

PyTorch入门实战教程

Leave a Reply

Your email address will not be published. Required fields are marked *

1条评论

  1. 最后预测部分:predict_word = idx_to_word[predict_label.data[0][0]],报错了:KeyError: tensor(81);改为predict_word = idx_to_word[int(predict_label.data[0][0])] 可行。

返回顶部