poem_generator

Generate Chinese poem automatically.

Requirements

Usage

For the first time, run the commands below to init.

python preprocess.py
python get_collocations.py
python get_topic.py
python get_start_words.py

Next time, just run this command.

python index.py

Data

In ./data folder, there is a corpus file "唐诗语料库.txt", and some data files will be generated here.