文件名称：CoupletAI:基于CNN + Bi-LSTM + Attention的自动对对联系统
文件大小：33KB
文件格式：ZIP
更新时间：2024-03-31 10:12:46
系统开源
对联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
【文件预览】：
CoupletAI-master
----module()
--------tokenizer.py(2KB)
--------metric.py(998B)
--------model.py(5KB)
--------__init__.py(1KB)
----docker()
--------Dockerfile(340B)
----webdemo.py(1KB)
----main.py(7KB)
----LICENSE(1KB)
----README.md(2KB)
----docs()
--------web_demo.png(4KB)
--------webapi_demo.png(10KB)
--------terminal_demo.png(6KB)
----clidemo.py(1KB)
----templates()
--------index.html(654B)
----.gitignore(1KB)
----preprocess.py(4KB)