Advertisement
TSKumarage

Untitled

Jun 5th, 2020
879
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.44 KB | None | 0 0
  1. def _get_lstm_features(self, sentence):
  2.         self.hidden = self.init_hidden()
  3.         embeds = self.word_embeds(sentence).view(len(sentence), 1, -1)
  4.         atten_weights = self.self_attn(embeds)
  5.         new_embeds = embeds x atten_weights
  6.  
  7.         lstm_out, self.hidden = self.lstm(new_embeds, self.hidden)
  8.         lstm_out = lstm_out.view(len(sentence), self.hidden_dim)
  9.         lstm_feats = self.hidden2tag(lstm_out)
  10.         return lstm_feats
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement