使用了:python+pytorch+Transformer+BiLSTM+ATTN+CNN
自己出上联,AI可以对出下联
运行 preprocess.py 进行数据预处理(已经训练好了,可选择忽略)
运行 main.py [-m model type] 进行训练(已经训练好了,可选择忽略)
运行 clidemo.py <-p model path> 可在控制台进行AI对对联
运行 webdemo.py 可在Web端进行AI对对联
命令行参数的详细说明见文件内,你也可以在 module/model.py 中定义你自己的模型。
Using Docker:docker pull wisedoge/coupletai