登录 注册
当前位置:主页 > 资源下载 > 28 > CoupletAI:一个运用CNN、Bi-LSTM及Attention机制的自动对联生成系统

CoupletAI:一个运用CNN、Bi-LSTM及Attention机制的自动对联生成系统

  • 更新:2024-07-27 19:15:56
  • 大小:33KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:数据集 - 行业研究
  • 格式:ZIP

资源介绍

对联AI 用PyTorch实现的自动对对联系统,支持多种模型。一般而言,给定一句话生成另一句话是序列生成问题,本项目根据上下联字数不同的特点将其转化为序列标注问题,即用下联去标注上联。 依存关系 python 3.6+ pytorch 1.2+ 烧瓶(可选) 数据集 数据集包含70多万条对联数据(26MB),下载请,或者(提取码:wude)。 用法 将下载到的数据集解压到当前目录(解压后的文件夹名称为couplet ) 运行preprocess.py进行数据预 运行main.py [-m model type]进行训练 运行clidemo.py <-p model path>可在控制台进行AI对对联 运行webdemo.py 可在Web端进行AI对对联 命令行参数的详细说明见文件内,你也可以在module/model.py中定义你自己的模型。 使用Docker