ptb_word_test.py 文件源码

def main():
    # ?? ???,???,???
    raw_data = reader.ptb_raw_data('/home/feizhihui/MyData/dataset/PTB')
    train_data, valid_data, test_data, vocab = raw_data
    # ??????config{small,medium,large,or test}
    # ??2?????,?????config??????
    config = TestConfig()
    with tf.Graph().as_default(), tf.Session() as session:
        with tf.variable_scope("model"):
            m = PTBModel(is_training=True, config=config)

        saver = tf.train.Saver()
        saver.restore(session, "PTB_Model/PTB_Variables.ckpt")

        print(vocab)
        rvocab = dict(zip(vocab.values(), vocab.keys()))
        lstm_state_value = session.run(m.initial_state)
        while True:
            lstm_state_value = predict(session, m, vocab, rvocab, lstm_state_value)