mirror of
https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git
synced 2021-05-12 18:32:23 +03:00
Merge pull request #30 from kelvinkoh0308/patch-7
Update lstm_sentiment_analysis_layer.py
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user