Merge pull request #30 from kelvinkoh0308/patch-7

Update lstm_sentiment_analysis_layer.py
This commit is contained in:
Jackie Loong
2019-09-17 20:46:26 +10:00
committed by GitHub

View File

@@ -69,7 +69,7 @@ class MyRNN(keras.Model):
x = self.embedding(x)
# rnn cell compute
# x: [b, 80, 100] => [b, 64]
x = self.rnn(x)
x = self.rnn(x,training=training)
# out: [b, 64] => [b, 1]
x = self.outlayer(x)