Skip to content
Snippets Groups Projects
Commit ebbab31e authored by Recolic K's avatar Recolic K
Browse files

fix issues

parent d2415902
No related branches found
No related tags found
No related merge requests found
......@@ -46,6 +46,7 @@ class LSTM_Shortterm_Predictor(nn.Module):
# The LSTM hidden states
# with dimensionality hidden_dim.
self.lstm = nn.LSTM(self.lstm_idim, self.lstm_odim)
# TODO: this model is naive. solve the bug in dataset processing.
# The linear layer that maps from hidden state space to tag space
self.out = nn.Linear(self.lstm_odim, 1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment