Advertisement
Guest User

Untitled

a guest
Apr 5th, 2020
289
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.89 KB | None | 0 0
  1. def evaluate(encoder, decoder, sentence, max_length=MAX_LENGTH):
  2.     with torch.no_grad():
  3.         input_tensor = tensorFromSentenceWithChar(input_lang, sentence)
  4.         input_length = input_tensor.size()[0]
  5.         encoder_hidden = encoder.initHidden()
  6.  
  7.         encoder_outputs = torch.zeros(max_length, encoder.hidden_size, device=device)
  8.  
  9.         for ei in range(input_length):
  10.             encoder_output, encoder_hidden = encoder(input_tensor[ei],
  11.                                                      encoder_hidden)
  12.             encoder_outputs[ei] += encoder_output[0, 0]
  13.  
  14.         decoder_input = torch.tensor([[SOS_token]], device=device)  # SOS
  15.  
  16.         decoder_hidden = encoder_hidden
  17.  
  18.         decoded_words = []
  19.         decoder_attentions = torch.zeros(max_length, max_length)
  20.      
  21.        
  22.         return beam_decode(decoder_input, decoder_hidden, encoder_outputs, decoder), None
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement