
CoupletAI是一个利用卷积神经网络(CNN)、双向长短期记忆网络(Bi-LSTM)和注意力机制的自动创作对子系统。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
对联AI系统,采用PyTorch实现,构建了自动生成对联的系统,并支持多种模型类型。通常情况下,将一句话转换为另一句话的处理被视为序列生成任务。本项目则针对上联与下联字数差异的特性,巧妙地将其转化为序列标注问题,具体而言,是通过下联来辅助标注上联。该项目依赖于Python 3.6及以上版本,以及PyTorch 1.2及以上版本。为了提供更灵活的运行环境,建议使用烧瓶(可选)。数据集方面,我们提供了包含70多万条对联数据的资源(文件大小约为26MB),您可以选择直接下载或通过提供的提取码进行下载。数据集已解压后放置在当前目录,文件夹命名为“couplet”。随后,运行preprocess.py脚本进行数据预处理。接着执行main.py脚本,并使用[-m model type]参数指定模型类型以进行训练。通过clidemo.py脚本可以在控制台体验AI对联功能;同时,您可以使用
全部评论 (0)
还没有任何评论哟~


