model_test.py 文件源码

def _createTestEvalModel(self, m_creator, hparams, sess):
    eval_mode = tf.contrib.learn.ModeKeys.EVAL
    eval_iterator, src_vocab_table, tgt_vocab_table = common_test_utils.create_test_iterator(
        hparams, eval_mode)
    eval_m = m_creator(
        hparams,
        eval_mode,
        eval_iterator,
        src_vocab_table,
        tgt_vocab_table,
        scope='dynamic_seq2seq')
    sess.run(tf.tables_initializer())
    sess.run(eval_iterator.initializer)
    return eval_m