# CoupletAI
用 PyTorch 实现的自动对对联系统,支持多种模型。一般来讲,给定一句话生成另一句话是序列生成问题,本项目根据上下联字数相等的特点将其转化为序列标注问题,即用下联去标注上联。
## Dependencies
* python 3.6+
* pytorch 1.2+
* flask (optional)
## Dataset
数据集包含70多万条对联数据(26MB),下载请[点击这里](https://github.com/wb14123/couplet-dataset/releases/download/1.0/couplet.tar.gz),或者[百度云](https://pan.baidu.com/s/1Zqnqq0VqZxv2c4jTNlZJGQ)(提取码: wude)。
## Usage
* 将下载到的数据集解压到当前目录(解压后的文件夹名称为`couplet`)
* 运行 `preprocess.py` 进行数据预处理
* 运行 `main.py [-m model type]` 进行训练
* 运行 `clidemo.py <-p model path>` 可在控制台进行AI对对联
* 运行 `webdemo.py <model path>` 可在Web端进行AI对对联
命令行参数的详细说明见文件内,你也可以在 `module/model.py` 中定义你自己的模型。
## Using Docker
```docker pull wisedoge/coupletai```
## Results Show
| # | 对联 |
| ------------ | ---------------------------------- |
| 上联 | 放不开眼底乾坤,何必登斯楼把酒 |
| 下联 | 吞得尽胸中云梦,方许对古人言诗 |
| AI预测的下联 | 抛难在胸中日月,不然看此水凭诗 |
| 上联 | 春暮偶登楼,上下鱼龙,应惜满湖绿水 |
| 下联 | 酒醉休说梦,关山戎马,未如一枕黄梁 |
| AI预测的下联 | 秋寒常入酒,东来风水,更喜一岸红山 |
| 上联 | 一器成名只为茗 |
| 下联 | 悦来客满是茶香 |
| AI预测的下联 | 三年有梦不因诗 |
| 上联 | 春夜灯花,几处笙歌腾朗月 |
| 下联 | 良宵美景,万家箫管乐丰年 |
| AI预测的下联 | 秋天月雨,一时风雨点清风 |
| 上联 | 一曲笙歌春似海 |
| 下联 | 千门灯火夜如年 |
| AI预测的下联 | 三年灯色梦如山 |
## Screenshot
### 命令行运行
![Terminal Demo](docs/terminal_demo.png)
### 网页运行
![Web Demo](docs/web_demo.png)
### Web API
![Web API](docs/webapi_demo.png)
没有合适的资源?快使用搜索试试~ 我知道了~
新年用Python与人工智能一起写春节对联
共65个文件
sample:11个
py:8个
pyc:6个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
5星 · 超过95%的资源 3 下载量 32 浏览量
2022-01-26
12:51:47
上传
评论
收藏 86MB RAR 举报
温馨提示
使用了:python+pytorch+Transformer+BiLSTM+ATTN+CNN 自己出上联,AI可以对出下联 运行 preprocess.py 进行数据预处理(已经训练好了,可选择忽略) 运行 main.py [-m model type] 进行训练(已经训练好了,可选择忽略) 运行 clidemo.py <-p model path> 可在控制台进行AI对对联 运行 webdemo.py <model path> 可在Web端进行AI对对联 命令行参数的详细说明见文件内,你也可以在 module/model.py 中定义你自己的模型。 Using Docker:docker pull wisedoge/coupletai
资源推荐
资源详情
资源评论
收起资源包目录
CoupletAI.rar (65个子文件)
CoupletAI
.gitignore 1KB
.git
info
exclude 240B
objects
pack
pack-e4ad54bd116490e7fd11ec633d9f714fd7b992ed.pack 58KB
pack-e4ad54bd116490e7fd11ec633d9f714fd7b992ed.idx 5KB
info
HEAD 23B
description 73B
packed-refs 235B
branches
config 266B
index 1KB
refs
tags
remotes
origin
HEAD 32B
heads
master 41B
hooks
commit-msg.sample 896B
pre-receive.sample 544B
fsmonitor-watchman.sample 3KB
pre-rebase.sample 5KB
prepare-commit-msg.sample 1KB
update.sample 4KB
pre-push.sample 1KB
pre-commit.sample 2KB
post-update.sample 189B
applypatch-msg.sample 478B
pre-applypatch.sample 424B
logs
HEAD 185B
refs
remotes
origin
HEAD 185B
heads
master 185B
templates
index.html 654B
output
Transformer_10.bin 6.95MB
Transformer_20.bin 6.95MB
dataset
test.pkl 2.11MB
vocab.pkl 213KB
train.pkl 405.61MB
LICENSE 1KB
module
__init__.py 1KB
tokenizer.py 2KB
__pycache__
metric.cpython-37.pyc 1KB
tokenizer.cpython-37.pyc 3KB
model.cpython-37.pyc 6KB
__init__.cpython-37.pyc 957B
metric.py 998B
model.py 5KB
runs
events.out.tfevents.1643163644.8e319d9c2b12.310.0 9KB
README.md 2KB
.idea
.gitignore 50B
misc.xml 310B
workspace.xml 4KB
vcs.xml 185B
CoupletAI.iml 336B
inspectionProfiles
profiles_settings.xml 174B
modules.xml 277B
__pycache__
main.cpython-36.pyc 7KB
main.cpython-37.pyc 7KB
preprocess.py 4KB
couplet
vocabs 36KB
train
out.txt 28.22MB
in.txt 28.22MB
test
out.txt 151KB
in.txt 151KB
couplet.tar.gz 26.14MB
main.py 7KB
docs
terminal_demo.png 6KB
web_demo.png 4KB
webapi_demo.png 10KB
webdemo.py 1KB
clidemo.py 1KB
docker
Dockerfile 340B
共 65 条
- 1
资源评论
- xjng2023-01-16怎么能有这么好的资源!只能用感激涕零来形容TAT...
- weixin_513468522022-12-16资源很实用,内容详细,值得借鉴的内容很多,感谢分享。
- qq_399799902022-04-27用户下载后在一定时间内未进行评价,系统默认好评。
中杯可乐多加冰
- 粉丝: 5w+
- 资源: 60
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- keil2 + proteus + 8051.exe
- 1961ee27df03bd4595d28e24b00dde4e_744c805f7e4fb4d40fa3f695bfbab035_8(1).c
- mediapipe-0.9.0.1-cp37-cp37m-win-amd64.whl.zip
- windows注册表编辑工具
- mediapipe-0.9.0.1-cp37-cp37m-win-amd64.whl.zip
- 校园通行码预约管理系统20240522075502
- 车类型数据集6250张VOC+YOLO格式.zip
- The PyTorch implementation of STGCN.STGCN-main.zip
- 092300108.cpp
- 车类型数据集6000张VOC+YOLO格式.zip
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功