Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def _get_lstm_features(self, sentence):
- self.hidden = self.init_hidden()
- embeds = self.word_embeds(sentence).view(len(sentence), 1, -1)
- atten_weights = self.self_attn(embeds)
- new_embeds = embeds x atten_weights
- lstm_out, self.hidden = self.lstm(new_embeds, self.hidden)
- lstm_out = lstm_out.view(len(sentence), self.hidden_dim)
- lstm_feats = self.hidden2tag(lstm_out)
- return lstm_feats
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement