用 PyTorch 实现的自动对对联系统，支持多种模型。一般来讲，给定一句话生成另一句话是序列生成问题，本项目根据上下联字数相等的特点将其转化为序列标注问题，即用下联去标注上联。

• python 3.6+
• pytorch 1.2+
• flask (optional)

数据集包含70多万条对联数据(26MB)，下载请点击这里，或者百度云(提取码: wude)。

• 将下载到的数据集解压到当前目录（解压后的文件夹名称为couplet）
• 运行 preprocess.py 进行数据预处理
• 运行 main.py [-m model type] 进行训练
• 运行 clidemo.py <-p model path> 可在控制台进行AI对对联
• 运行 webdemo.py <model path> 可在Web端进行AI对对联

命令行参数的详细说明见文件内，你也可以在 module/model.py 中定义你自己的模型。

docker pull wisedoge/coupletai