Skip to content

moon-hotel/TransformerCouplet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A Transformer Framework Based Couplet Task

一个基于Transformer网络结构的对联生成模型

1. 环境准备

  • Python==3.x
  • PyTorch>=1.5.0
  • torchtext==0.6.0

2. 使用方法

2.1 训练

直接执行如下命令即可进行模型训练:

python train.py

训练过程:

-- INFO: Epoch: 0, Batch[29/3010], Train loss : 8.965, Train acc: 0.094
-- INFO: Epoch: 0, Batch[59/3010], Train loss : 8.618, Train acc: 0.098
-- INFO: Epoch: 0, Batch[89/3010], Train loss : 8.366, Train acc: 0.099
-- INFO: Epoch: 0, Batch[119/3010], Train loss : 8.137, Train acc: 0.109
...

学习率变化:

2.2 预测(inference)

直接运行如下命令即可:

python couplet.py

示例结果:

上联晚风摇树树还挺
 AI: 朝露沾花花更红
下联晨露润花花更红

上联忽忽几晨昏离别间之疾病间之不及终年同静好
 AI茕茕小儿女孱羸若此娇憨若此更烦二老费精神
下联茕茕小儿女孱羸若此娇憨若此更烦二老费精神

上联风声雨声读书声声声入耳
 AI山色水色烟霞色色色宜人
下联家事国事天下事事事关心

上联上海自来水来自海上
 AI中山落叶松叶落山中

3. 结果

bleu评测结果

About

A Transformer Framework Based Couplet Task

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages