当前位置:   article > 正文

使用RNN联合注意力机制实现机器翻译_rnn机器翻译

rnn机器翻译

 https://zhuanlan.zhihu.com/p/28834212 

具体来自这一篇文章的指导

一、相关使用的查漏补缺:
1.其中的两种神奇的处理字符的操作:

2.关于nn.GRU()的参数解释和用法:

http://t.csdn.cn/30PZL

这篇文章讲得很清楚,需要用来预测的话看这篇也可以http://t.csdn.cn/VseAV

这里重点讲述以下它的参数的含义:

(1)输入参数:

(2)输出参数:

(3)用法示例:

3.关于nn.embedding的用法:

(1)就我看来,简单一点,就是将最里面的那个维度的数据由input_size->转化为 hidden_size,

然而并不是呢,看下面

(2)其他文章都没有chatGPT讲得清楚:

 

 

4.encoder 和attn_decoder的传参方式:

5.torch.data.topk()这个函数的作用:

听名字就知道这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。

6.nn.nllloss 和 crossEntropyLoss的区别:

二、这是我在kaggle运行的第一个project:-_-

虽然kaggle一周只能用30个小时的GPU P100,但是,只要用的时候开,基本用不完,而且比那个Colab快多了,嘻嘻,重点是不会自动断掉,非常虚浮

 1.关于Kaggle的使用经验:

(1)首先,登录之后,点击到code页面,添加新的notebook,就可以开始写代码了

(2)关于上传文件,我这一次是选择的从本地进行上传的方式,建议上传的时候关掉vpn比较快,然后再打开,注意,最好选择拖动的方式,可以直接上传“整个文件夹”,并给整个文件放到一个文件夹目录,这个目录需要自己命名,注意,.ipynb文件没啥用

(3)特别经验,有时候文件路径老是很让人困惑,所以以下tips:

  1. import os
  2. print(os.path.exists('./kaggle/input/main-data/RNN_for_translate/data/eng-fra.txt'))

如果文件存在,该命令会输出True;如果文件不存在,会输出False

2.关于具体的代码和注释:

  1. #引入相关的库内容:
  2. from __future__ import unicode_literals, print_function, division
  3. from io import open # 处理文件
  4. import unicodedata # 处理unicode字符相关事项
  5. import string
  6. import re # 正则表达式相关
  7. import random
  8. import torch
  9. import torch.nn as nn
  10. from torch.autograd import Variable
  11. from torch import optim
  12. import torch.nn.functional as F
  13. use_cuda = torch.cuda.is_available() # 如果您的计算机支持cuda,则优先在cuda下运行
  14. print(use_cuda) #看来本王的笔记本并不支持cuda
  1. #下载数据并且对数据进行处理:
  2. #把文本文件一次性读入内存
  3. #对文本中出现的一些特殊字符进行适当处理
  4. #分别建立“英语”、“法语”语料库(词向量),并对两库进行一定的修饰,剔除不常用的词、并剔除两库中使用不常用词的句子。
  5. #提供后续程序需要的一些功能接口
  6. #声明起始和结束占用符
  7. SOS_token = 0
  8. EOS_token = 1
  9. #我们构建一个类Lang,来全权处理文本数据相关的操作:
  10. class Lang: #这个注释挺详细的,而且这个部分的思想和c++的类设计很相似
  11. def __init__(self, name): #初始化函数,传递一个name参数
  12. self.name = name
  13. self.word2index = {} # 单词对应的在字典里的索引号
  14. self.word2count = {} # 记录某一个单词在语料库里出现的次数
  15. self.index2word = {0: "SOS", 1: "EOS"} # 索引对应的单词
  16. self.n_words = 2 # Count SOS and EOS # 语料库里拥有的单词数量
  17. def addSentence(self, sentence): # 往语料库里增加一句话:扩充语料库 ,参数是这句话
  18. for word in sentence.split(' '): # 要增加的一句话是以空格来分割不同的单词
  19. self.addWord(word) # 把单词一个个加入语料库
  20. def addWord(self, word): # 把单词加入到语料库中具体要做的事情
  21. if word not in self.word2index: # 对于语料库中不存在的新词
  22. self.word2index[word] = self.n_words # 索引号依据先来后到的次序分配
  23. self.word2count[word] = 1 # 更新该次的出现次数
  24. self.index2word[self.n_words] = word # 同时更新该字典
  25. self.n_words += 1
  26. else:
  27. self.word2count[word] += 1 # 对于已存在于语料库中的词,仅增加其出现次数。
  1. #文中还提供了两个辅助方法,来将unicode字符转化为ascii字符,
  2. #同时对英语、法语句子中存在一些大小写、缩写、连写、特殊字符等现象进行的规范化处理:
  3. # Turn a Unicode string to plain ASCII, thanks to
  4. # http://stackoverflow.com/a/518232/2809427
  5. def unicodeToAscii(s):
  6. return ''.join(
  7. c for c in unicodedata.normalize('NFD', s)
  8. if unicodedata.category(c) != 'Mn'
  9. )
  10. # Lowercase, trim, and remove non-letter characters
  11. def normalizeString(s):
  12. s = unicodeToAscii(s.lower().strip())
  13. s = re.sub(r"([.!?])", r" \1", s)
  14. s = re.sub(r"[^a-zA-Z.!?]+", r" ", s)
  15. return s
  1. #读取文件中的内容,形成一系列语句对,并构建两个Lang对象
  2. def readLangs(lang1, lang2, reverse=False): # lang1,lang2仅是字符串,代表对应的语言
  3. print("Reading lines...")
  4. # Read the file and split into lines
  5. lines = open('data/%s-%s.txt' % (lang1, lang2), encoding='utf-8').\
  6. read().strip().split('\n')
  7. # 把文本文件变为语句对列表
  8. # Split every line into pairs and normalize ,这里的结果是,将英文 和 法文 分开成对pair,然后构建一个元素是pair的列表叫“pairs”
  9. pairs = [[normalizeString(s) for s in l.split('\t')] for l in lines]
  10. # 提供一个反向的操作,即原来是英文->法语,使用reverse后则为法语->英语
  11. # Reverse pairs, make Lang instances
  12. if reverse:
  13. pairs = [list(reversed(p)) for p in pairs]
  14. input_lang = Lang(lang2)
  15. output_lang = Lang(lang1)
  16. else:
  17. input_lang = Lang(lang1)
  18. output_lang = Lang(lang2)
  19. return input_lang, output_lang, pairs #返回一个lang1和lang2对象,以及生成的语句对pairs
  1. #以下提供了一些辅助方法,用于从总的文本数据中筛选出感兴趣的数据来进行训练,
  2. #读者可以根据自己的兴趣决定是否使用下面这两个方法:
  3. MAX_LENGTH = 10
  4. eng_prefixes = (
  5. "i am ", "i m ",
  6. "he is", "he s ",
  7. "she is", "she s",
  8. "you are", "you re ",
  9. "we are", "we re ",
  10. "they are", "they re "
  11. )
  12. #过滤pair的函数p[0] 和 p[1]的长度 都要 <10 , 且 p[1]????(我感觉是p[0])必须以上面这些英语句子开头 ,如果reverse了,就没问题
  13. def filterPair(p): # 作者仅对训练数据中句子长度都小于10,且以一定字符串开头的英文句子感兴趣
  14. return len(p[0].split(' ')) < MAX_LENGTH and len(p[1].split(' ')) < MAX_LENGTH and \
  15. p[1].startswith(eng_prefixes)
  16. def filterPairs(pairs): # 从所有pairs中选出作者感兴趣的pair
  17. return [pair for pair in pairs if filterPair(pair)]
  1. #开始准备数据:
  2. def prepareData(lang1, lang2, reverse=False): #传递2个语言的string参数, 以及是否reverse
  3. # 构建两个语料库
  4. input_lang, output_lang, pairs = readLangs(lang1, lang2, reverse) #调用上述的readLangs得到 pairs数组
  5. print("Read %s sentence pairs" % len(pairs)) #输出总共多少个pair对
  6. pairs = filterPairs(pairs) # 筛选感兴趣的语句对
  7. print("Trimmed to %s sentence pairs" % len(pairs)) #筛选之后 剩下的pair对数目
  8. print("Counting words...") # 统计词频
  9. for pair in pairs:
  10. input_lang.addSentence(pair[0]) #pair[0]应该是输入的,也就是英文,添加到input_lang这个对象中,之后也处理了word
  11. output_lang.addSentence(pair[1]) #感觉这样的话,句子和句子之间的关系就对不上了,,,???只是挨个将词放进去
  12. print("Counted words:")
  13. print(input_lang.name, input_lang.n_words) #输出input_lang 和 output_lang中的word数目
  14. print(output_lang.name, output_lang.n_words)
  15. return input_lang, output_lang, pairs #返回input_lang和 out_lang对象,以及这个pairs数组
  16. input_lang, output_lang, pairs = prepareData('eng', 'fra', True) #创建lang1,lang2,pairs,原来是进行了reverse反转
  17. print(random.choice(pairs)) #随机输出pairs中的一个pair对
  1. #定义RNN类型的 Encoder
  2. class EncoderRNN(nn.Module):
  3. def __init__(self, input_size, hidden_size, n_layers=1):
  4. super(EncoderRNN, self).__init__()
  5. self.n_layers = n_layers # 可以使用多层gru操作,默认只使用一层
  6. self.hidden_size = hidden_size # 隐藏层的尺寸,如何设定参考后续代码
  7. self.embedding = nn.Embedding(input_size, hidden_size) #input_size个词汇,每个词汇用hidden_size维度的向量表示
  8. self.gru = nn.GRU(hidden_size, hidden_size) # 记忆结构
  9. def forward(self, input, hidden):
  10. embedded = self.embedding(input).view(1, 1, -1) #先将输入embedding到embedded。然后调整形状为3维的tensor -1*1*all
  11. output = embedded #将embedded传给output
  12. for i in range(self.n_layers):
  13. output, hidden = self.gru(output, hidden) #通过n_layers层的gru,然后得到输出output 和 状态hidden
  14. return output, hidden #参看那个博客的gru参数以及用法,你就明白了,
  15. #参数embedded 1*1*all , hidden _就是那个h_0 1*1*hidden_size
  16. #参数output 1*1*hidden_size , hidden 1*1*hidden_size
  17. def initHidden(self):
  18. result = Variable(torch.zeros(1, 1, self.hidden_size)) #就是得到上面的那个1*1*hidden_size -?上面没有调用?还是之后调用?下面有说
  19. if use_cuda:
  20. return result.cuda()
  21. else:
  22. return result
  1. #Decoder_version1 __普通版本的RNN_decoder_model
  2. class DecoderRNN(nn.Module):
  3. def __init__(self, hidden_size, output_size, n_layers=1):
  4. super(DecoderRNN, self).__init__()
  5. self.n_layers = n_layers
  6. self.hidden_size = hidden_size
  7. self.embedding = nn.Embedding(output_size, hidden_size) #定义embedding函数,词汇表的大小output_size,每个词汇的向量长度hidden_size
  8. self.gru = nn.GRU(hidden_size, hidden_size) #gru的input参数 和 h_0参数
  9. self.out = nn.Linear(hidden_size, output_size) #从hidden_size维度 到 output_size维度的Linear层
  10. # 由于输出是语料库中词语的概率,选最大概率的索引对应的词,
  11. # 所以需要一个类softmax操作
  12. self.softmax = nn.LogSoftmax() #由于是一个多分类问题,最后通过一次softmax(前面还有一个log操作)
  13. def forward(self, input, hidden): #具体的Decoder运行过程
  14. output = self.embedding(input).view(1, 1, -1) #还是将input转化为1*1*all3维空间
  15. for i in range(self.n_layers): #多少层捏
  16. output = F.relu(output) #通过1个relu
  17. output, hidden = self.gru(output, hidden) #再和h_0一起通过gru
  18. output = self.softmax(self.out(output[0])) #最后对output的有用的那层数据进行Linear+softmax
  19. return output, hidden #返回output 和 h_n
  20. def initHidden(self): #我甚至怀疑这个函数是自动调用的了,产生了一个1*1*hidden_size作为h_0输入量
  21. result = Variable(torch.zeros(1, 1, self.hidden_size))
  22. if use_cuda:
  23. return result.cuda()
  24. else:
  25. return result
  1. #加上self_attention版本的RNN_Decoder
  2. class AttnDecoderRNN(nn.Module):
  3. def __init__(self, hidden_size, output_size, n_layers=1, dropout_p=0.1, max_length=MAX_LENGTH):
  4. #对于解码器来说,最重要的两个参数是 隐藏状态的尺寸 和 输出的尺寸大小,这两者主要决定了解码器的参数规模
  5. super(AttnDecoderRNN, self).__init__()
  6. self.hidden_size = hidden_size #先把参数传进去
  7. self.output_size = output_size
  8. self.n_layers = n_layers
  9. self.dropout_p = dropout_p
  10. self.max_length = max_length # 还有一个句子最大长度参数
  11. self.embedding = nn.Embedding(self.output_size, self.hidden_size) #设置embedding函数
  12. self.attn = nn.Linear(self.hidden_size * 2, self.max_length) #设置attn这个linear函数
  13. self.attn_combine = nn.Linear(self.hidden_size * 2, self.hidden_size)#设置attn_combine这个linear函数
  14. self.dropout = nn.Dropout(self.dropout_p) #设置dropout函数
  15. self.gru = nn.GRU(self.hidden_size, self.hidden_size) #设置gru函数
  16. self.out = nn.Linear(self.hidden_size, self.output_size) #设置out输出linear函数
  17. def forward(self, input, hidden, encoder_output, encoder_outputs):#整个Decoder运行过程
  18. embedded = self.embedding(input).view(1, 1, -1) #进行embedding将词汇转换为向量序列后,准换为1*1*all的形状
  19. embedded = self.dropout(embedded) #调用一次dropout
  20. attn_weights = F.softmax(
  21. self.attn(torch.cat((embedded[0], hidden[0]), 1))) #attn_weights得到的过程:通过embedded和h_n的cat后通过attn线性层+softmax
  22. attn_applied = torch.bmm(attn_weights.unsqueeze(0),
  23. encoder_outputs.unsqueeze(0)) #通过bmm对attn_weights 和 encoder_outputs这2批矩阵进行对应相乘
  24. output = torch.cat((embedded[0], attn_applied[0]), 1) #将embedded和attn_applied相连
  25. output = self.attn_combine(output).unsqueeze(0) #通过线性层
  26. for i in range(self.n_layers):
  27. output = F.relu(output)
  28. output, hidden = self.gru(output, hidden) #通过n_layers次的relu和gru
  29. output = F.log_softmax(self.out(output[0])) #通过log_softmax
  30. return output, hidden, attn_weights #返回output 和 h_n 还有attn_weights
  31. def initHidden(self): #自动生成h_0,嘻嘻
  32. result = Variable(torch.zeros(1, 1, self.hidden_size))
  33. if use_cuda:
  34. return result.cuda()
  35. else:
  36. return result
  1. #数据准备部分:
  2. def indexesFromSentence(lang, sentence): #这里应该是输入一个lang的句子,然后返回这个句子的index序列
  3. return [lang.word2index[word] for word in sentence.split(' ')]
  4. def variableFromSentence(lang, sentence):
  5. indexes = indexesFromSentence(lang, sentence) #调用上面那个函数,获得这个句子的序列
  6. indexes.append(EOS_token) #在这个句子的最后加上EOS_token(1还是0来着。。。)这个值
  7. result = Variable(torch.LongTensor(indexes).view(-1, 1)) #将这个indexes序列转换为all*1维的数据
  8. if use_cuda:
  9. return result.cuda()
  10. else:
  11. return result
  12. def variablesFromPair(pair): #输入pair对,一个句子 -vs- 一个句子的那种
  13. input_variable = variableFromSentence(input_lang, pair[0])
  14. target_variable = variableFromSentence(output_lang, pair[1])
  15. return (input_variable, target_variable) #调用上面的那个函数,返回该种数据对
  1. #模型训练部分的代码- 阅读详细的注释:
  2. teacher_forcing_ratio = 0.5 #解释见后
  3. def train(input_variable, target_variable, encoder, decoder, encoder_optimizer, decoder_optimizer, criterion, max_length=MAX_LENGTH):
  4. # 初始化编码器的隐藏层状态
  5. encoder_hidden = encoder.initHidden() #果然,这个函数是用来初始化h_0的,嘻嘻
  6. # 清除编码器、解码器的梯度数据,准备接受下一次的梯度数据
  7. encoder_optimizer.zero_grad()
  8. decoder_optimizer.zero_grad()
  9. # 待翻译句子和已翻译句子的长度(即组成句子的词语的数量)
  10. input_length = input_variable.size()[0]
  11. target_length = target_variable.size()[0]
  12. # 建立一个编码器输出的PyTorch变量,注意命名是-s结尾,表示
  13. # 该变量保存了Encoder每一次中间状态数据,而不是最后一次中间状态。
  14. encoder_outputs = Variable(torch.zeros(max_length, encoder.hidden_size))
  15. # 如果使用cuda,则再包装一下
  16. encoder_outputs = encoder_outputs.cuda() if use_cuda else encoder_outputs
  17. loss = 0
  18. # 编码器的编码过程
  19. for ei in range(input_length):
  20. encoder_output, encoder_hidden = encoder(
  21. input_variable[ei], encoder_hidden)
  22. encoder_outputs[ei] = encoder_output[0][0]
  23. # 通过编码过程,得到了编码器的每一次中间状态数据
  24. # 给解码器准备最初的输入,是一个开始占位符
  25. decoder_input = Variable(torch.LongTensor([[SOS_token]]))
  26. decoder_input = decoder_input.cuda() if use_cuda else decoder_input
  27. # 解码器初始的输入就是编码器最后一次中间层状态数据
  28. decoder_hidden = encoder_hidden
  29. #终于明白下面的这个过程了,就是李宏毅上课讲到的,如果decoder中某个字符翻译错了,
  30. #到底下一次是用 正确的作为token输入呢 还是将错就错用 这个错误的输出作为输入呢?
  31. #下面是按照概率进行分2种进行的
  32. # 该变量表明是否在每一次输出时都是用目标正确输出来计算损失
  33. use_teacher_forcing = True if random.random() < teacher_forcing_ratio else False
  34. if use_teacher_forcing:
  35. # 条件为真时,使用正确的输出作为下一时刻解码器的输入来循环计算
  36. # Teacher forcing: Feed the target as the next input
  37. for di in range(target_length): #反正就是1个词的翻译,然后指导这个target这个label的长度 或者 输出得到EOS_token才退出decoder
  38. # decoder解码器具体实施的过程,确定其输出、隐藏层状态、以及注意力数据
  39. # decoder的forward方法会动态的确定decoder_attention数据
  40. decoder_output, decoder_hidden, decoder_attention = decoder(
  41. decoder_input, decoder_hidden, encoder_output, encoder_outputs)
  42. # 更新损失
  43. loss += criterion(decoder_output, target_variable[di])
  44. # 确定下一时间步的解码器输入
  45. decoder_input = target_variable[di] # Teacher forcing -这里就是用target作为下一次decoder的输入了
  46. else:
  47. # 条件不为真时,使用解码器自身预测的输出来作为下一时刻解码器的输入来循环计算
  48. # Without teacher forcing: use its own predictions as the next input
  49. for di in range(target_length):
  50. decoder_output, decoder_hidden, decoder_attention = decoder(
  51. decoder_input, decoder_hidden, encoder_output, encoder_outputs)
  52. #这里肯定就是获取到最大的概率的那个 位置的索引值
  53. topv, topi = decoder_output.data.topk(1) #获取前k个(前1)个元素的数值和位置索引
  54. ni = topi[0][0] #得到这个位置索引就是预测的词汇
  55. decoder_input = Variable(torch.LongTensor([[ni]])) #这里就是利用这次的预测 输出 作为下一次decoder的输入了
  56. decoder_input = decoder_input.cuda() if use_cuda else decoder_input
  57. loss += criterion(decoder_output, target_variable[di])
  58. if ni == EOS_token: #如果ni就是 EOS_token值了,就可以直接退出了
  59. break
  60. # 反向传递损失
  61. loss.backward()
  62. # 更新整个网络模型的参数
  63. encoder_optimizer.step()
  64. decoder_optimizer.step()
  65. # 该方法是训练过程,训练过程仅输出了训练的损失,并不提供翻译得到的句子,会有专门
  66. # 的方法来实施翻译过程。
  67. return loss.item() / target_length #返回训练中的平均loss数值
  1. #下面是2个用来计时的辅助函数:
  2. import time
  3. import math
  4. def asMinutes(s): #将输入数值转换为 几分 几秒返回
  5. m = math.floor(s / 60)
  6. s -= m * 60
  7. return '%dm %ds' % (m, s)
  8. def timeSince(since, percent): #
  9. now = time.time() #现在的时间
  10. s = now - since #相对的时间
  11. es = s / (percent) #。。。算了,先不管了,回头来看看
  12. rs = es - s
  13. return '%s (- %s)' % (asMinutes(s), asMinutes(rs))
  1. #这里才是正真的 train多次的 情况:
  2. def trainIters(encoder, decoder, n_iters, print_every=1000, plot_every=100, learning_rate=0.01): #每隔1000print一次,每隔100话一次
  3. start = time.time() #启动计时
  4. plot_losses = [] #保存需要绘制的loss
  5. print_loss_total = 0 #Reset every print_every 设置loss采样频率
  6. plot_loss_total = 0 #Reset every plot_every
  7. # 声明两个RNN的优化器
  8. encoder_optimizer = optim.SGD(encoder.parameters(), lr=learning_rate)
  9. decoder_optimizer = optim.SGD(decoder.parameters(), lr=learning_rate)
  10. # 得到训练使用的数据
  11. training_pairs = [variablesFromPair(random.choice(pairs))
  12. for i in range(n_iters)] #获取pairs总共多少对,然后每一对拆分为2个数组-感觉training_pair[2][n_iters][-]
  13. # 损失计算方法
  14. criterion = nn.NLLLoss() #一个少了softmax、log的crossEntropy
  15. # 循环训练,迭代的次数
  16. for iter in range(1, n_iters + 1): #总共n_iters对句子 , 果然之前定义的那个train函数只能用在1个train循环种
  17. training_pair = training_pairs[iter - 1] #获取输入句子的词汇数组 和 输出label的词汇数组
  18. input_variable = training_pair[0]
  19. target_variable = training_pair[1]
  20. loss = train(input_variable, target_variable, encoder,
  21. decoder, encoder_optimizer, decoder_optimizer, criterion) #对这一对句子input 和 label调用train得到loss
  22. print_loss_total += loss #总print_loss+total加上
  23. plot_loss_total += loss
  24. #每隔1000输出一次平均loss
  25. if iter % print_every == 0:
  26. print_loss_avg = print_loss_total / print_every #计算这一轮(1000个句子对)的平均loss
  27. print_loss_total = 0 #置零
  28. print('%s (%d %d%%) %.4f' % (timeSince(start, iter / n_iters),
  29. iter, iter / n_iters * 100, print_loss_avg)) #用来多少时间,计算了多少对,完成了百分之几,平均loss
  30. #每个100输出一次平均loss
  31. if iter % plot_every == 0:
  32. plot_loss_avg = plot_loss_total / plot_every #计算平均loss
  33. plot_losses.append(plot_loss_avg) #加到plot_loss_avg数组种,用于绘制 折线图
  34. plot_loss_total = 0 #置零,等下一轮用
  35. showPlot(plot_losses) #这个画图函数在下面会进行定义的
  1. import matplotlib.pyplot as plt
  2. import matplotlib.ticker as ticker
  3. import numpy as np
  4. def showPlot(points):
  5. plt.figure()
  6. fig, ax = plt.subplots()
  7. # this locator puts ticks at regular intervals
  8. loc = ticker.MultipleLocator(base=0.2)
  9. ax.yaxis.set_major_locator(loc)
  10. plt.plot(points) #反正就是绘制折线图,先不管了这里
  1. #evaluate 部分的代码:
  2. def evaluate(encoder, decoder, sentence, max_length=MAX_LENGTH):
  3. # 把sentence转化为网络可以接受的输入,同时初始化编码器的隐藏状态
  4. input_variable = variableFromSentence(input_lang, sentence) #获取输入句子的词汇数组 ,
  5. input_length = input_variable.size()[0]
  6. encoder_hidden = encoder.initHidden()
  7. # 准备编码器输出变量
  8. encoder_outputs = Variable(torch.zeros(max_length, encoder.hidden_size)) #获取一个max_length*hidden_size的数组
  9. encoder_outputs = encoder_outputs.cuda() if use_cuda else encoder_outputs
  10. # 得到编码的输出
  11. for ei in range(input_length): #???为什么,难道有多个是1个个词汇进行的encoder,what
  12. encoder_output, encoder_hidden = encoder(input_variable[ei],
  13. encoder_hidden)
  14. encoder_outputs[ei] = encoder_outputs[ei] + encoder_output[0][0]
  15. # 准备解码器输出的变量
  16. decoder_input = Variable(torch.LongTensor([[SOS_token]])) # SOS
  17. decoder_input = decoder_input.cuda() if use_cuda else decoder_input
  18. # 编码器和解码器之间的桥梁:Context
  19. decoder_hidden = encoder_hidden
  20. # 准备一个列表来保存网络预测的词语
  21. decoded_words = []
  22. # 准备一个变量保存解码过程中产生的注意力数据
  23. decoder_attentions = torch.zeros(max_length, max_length)
  24. # 解码过程,有一个最大长度限制
  25. for di in range(max_length):
  26. decoder_output, decoder_hidden, decoder_attention = decoder(
  27. decoder_input, decoder_hidden, encoder_output, encoder_outputs)
  28. decoder_attentions[di] = decoder_attention.data
  29. topv, topi = decoder_output.data.topk(1)
  30. ni = topi[0][0]
  31. if ni == EOS_token:
  32. decoded_words.append('<EOS>')
  33. break
  34. else:
  35. decoded_words.append(output_lang.index2word[ni])
  36. # 解码器的输出作为其输入 , 这里就不用 teacher forcing了
  37. decoder_input = Variable(torch.LongTensor([[ni]]))
  38. decoder_input = decoder_input.cuda() if use_cuda else decoder_input
  39. # 返回预测的单词,以及注意力机制(供分析注意力机制)
  40. return decoded_words, decoder_attentions[:di + 1]
  1. #观察其中某10个句子的预测情况
  2. def evaluateRandomly(encoder, decoder, n=10):
  3. for i in range(n):
  4. pair = random.choice(pairs) #随机选1对句子
  5. print('>', pair[0])
  6. print('=', pair[1])
  7. output_words, attentions = evaluate(encoder, decoder, pair[0]) #对pair进行evalute预测
  8. output_sentence = ' '.join(output_words) #输出的翻译结果存到output_sentence,并输出
  9. print('<', output_sentence)
  10. print('')
  1. #下面的三行代码在之前介绍过
  2. hidden_size = 256
  3. encoder1 = EncoderRNN(input_lang.n_words, hidden_size)
  4. attn_decoder1 = AttnDecoderRNN(hidden_size, output_lang.n_words,1, dropout_p=0.1)
  5. # 支持cuda计算
  6. if use_cuda:
  7. encoder1 = encoder1.cuda()
  8. attn_decoder1 = attn_decoder1.cuda()
  9. # 核心的训练代码仅此一句
  10. trainIters(encoder1, attn_decoder1, 75000, print_every=5000)
  11. #自己笔记本的cpu上面根本跑不动,还是用kaggle试一试
  12. #之前在这里发生错误的原因,就是他们老是使用 老版本的loss.data[0]这个已经不能用了,而应该改用loss.item()
  1. #查看随机翻译10个句子的结果
  2. evaluateRandomly(encoder1, attn_decoder1)
  3. #这里有点奇怪,明明应该用test_Data,不过它还是在用同一个train_data
三、实验结果:
1.这就是在train上面的loss的变化结果了:(这个代码有值得后期继续改进的地方就是,整个data没有分出一部分来作为 validation,也没有利用validation进行翻译)

后期继续改进吧,。。。。。。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/348361
推荐阅读
  

闽ICP备14008679号