Advertisement
Guest User

Untitled

a guest
Apr 6th, 2020
244
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.95 KB | None | 0 0
  1. def indexesFromSentence(lang, sentence):
  2.     return [lang.word2index[word] for word in sentence.split(' ')]
  3.  
  4. def indexesFromSentenceWithChar(lang, sentence):
  5.     return [[lang.char2index[char] for char in word if char in lang.char2index] for word in sentence.split(' ')]
  6.  
  7. def tensorFromSentence(lang, sentence): # For Word
  8.     indexes = indexesFromSentence(lang, sentence)
  9.     indexes.append(EOS_token)
  10.     return torch.tensor(indexes, dtype=torch.long, device=device).view(-1, 1)
  11.  
  12. def tensorFromSentenceWithChar(lang, sentence):
  13.     indexes = indexesFromSentenceWithChar(lang, sentence)
  14.     indexes.append([EOS_token])
  15.     indexes = list(itertools.chain.from_iterable(indexes))
  16.     return torch.tensor(indexes, dtype=torch.long, device=device).view(-1, 1)
  17.  
  18. def tensorsFromPair(pair):
  19.     input_tensor = tensorFromSentenceWithChar(input_lang, pair[0])
  20.     target_tensor = tensorFromSentence(output_lang, pair[1])
  21.     return (input_tensor, target_tensor)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement