07-lstm-tensorflow.py 文件源码

def get_book(name, language):
    book = lzma.open('../data/{}-common.vpl.xz'.format(name), 'rt').read().splitlines()
    book = [
        [ language(w, 0) for w in l]
        + ([veczero] * (n_steps - len(l)))
        for l in book ]
    for verse in book:
        assert len(verse) <= n_steps, "n_steps should be at least {}".format(len(verse))
    return book