model.py 文件源码

def make_seq2seq_model(**kwargs):
  args = dict(encoder_cell=LSTMCell(10),
        decoder_cell=LSTMCell(20),
        vocab_size=10,
        embedding_size=10,
        attention=True,
        bidirectional=True,
        debug=False)
  args.update(kwargs)
  return Seq2SeqModel(**args)