Advertisement
Guest User

Untitled

a guest
Mar 23rd, 2019
78
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.80 KB | None | 0 0
  1. def seq2seq_model_builder(HIDDEN_DIM=300):
  2.  
  3. encoder_inputs = Input(shape=(MAX_LEN, ), dtype='int32',)
  4. encoder_embedding = embed_layer(encoder_inputs)
  5. encoder_LSTM = LSTM(HIDDEN_DIM, return_state=True)
  6. encoder_outputs, state_h, state_c = encoder_LSTM(encoder_embedding)
  7.  
  8. decoder_inputs = Input(shape=(MAX_LEN, ), dtype='int32',)
  9. decoder_embedding = embed_layer(decoder_inputs)
  10. decoder_LSTM = LSTM(HIDDEN_DIM, return_state=True, return_sequences=True)
  11. decoder_outputs, _, _ = decoder_LSTM(decoder_embedding, initial_state=[state_h, state_c])
  12.  
  13. # dense_layer = Dense(VOCAB_SIZE, activation='softmax')
  14. outputs = TimeDistributed(Dense(VOCAB_SIZE, activation='softmax'))(decoder_outputs)
  15. model = Model([encoder_inputs, decoder_inputs], outputs)
  16.  
  17. return model
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement