用RNN与LSTM网络原理进行唐诗生成.tar.gz

import argparse


def parse_args（）:
    parser = argparse.ArgumentParser（description=‘Intelligence Poem and Lyric Writer.‘）

    help_ = ‘you can set this value in terminal --write value can be poem or lyric.‘
    parser.add_argument（‘-w‘ ‘--write‘ default=‘poem‘ choices=[‘poem‘ ‘lyric‘] help=help_）

    help_ = ‘choose to train or generate.‘
    parser.add_argument（‘--train‘ dest=‘train‘ action=‘store_true‘ help=help_）
    parser.add_argument（‘--no-train‘ dest=‘train‘ action=‘store_false‘ help=help_）
    parser.set_defaults（train=True）

    args_ = parser.parse_args（）
    return args_


if __name__ == ‘__main__‘:
    args = parse_args（）
    if args.write == ‘poem‘:
        from inference import poems
        if args.train:
            poems.main（True）
        else:
            poems.main（False）
    elif args.write == ‘lyric‘:
        from inference import song_lyrics
        print（args.train）
        if args.train:
            song_lyrics.main（True）
        else:
            song_lyrics.main（False）
    else:
        print（‘[INFO] write option can only be poem or lyric right now.‘）