明年，我要用 AI 给全村写对联

一位名叫王斌的程序员，爬取和使用该对联数据集，用 GTX 1080 Ti 训练了 4 天，得到了一个自动对对联模型。效果如何？我们来测试一下。

测试地址：https://ai.binwang.me/couplet/

这位程序员还将该项目在 GitHub 开源：

https://github.com/wb14123/couplet-dataset

另外一位工程师，则基于 GPT2-Chinese 和该对联数据集，训练了一份对联领域的 GPT-2 模型，可自动生成对联。

它将其 server 版本对接到微信公众号 AINLP 后台，可以直接在对话框进行测试。

输入关键词「写对联」，可触发对联自动生成，例如输入「写对联金牛」，对联模型会基于「金牛」进行自动续写，给出以「金牛」开头的 3 个对联：

输入「对对联 + 一组上联」，可得到三组对联，比如：

目前看来，对于一些经典对联，AI 能做到对仗比较工整，且想象力丰富，对于同一上联，可对出多个不同下联；而对于一些不常在对联中出现的字词，AI 表现有些随意牵强。当然，让 AI 对「人工智能」什么的，也的确是有些为难人家了。

华为团队开发的 AI 对联，也是以 Couplet-dataset 对联数据集作为训练数据。该模型可以根据给定主题，生成对联。 以下分别是输入「新春快乐」、「平安健康」、「一夜暴富」主题的生成结果， 可滑动查看：

明年，你要不要也训练一个自己的 AI 写春联模型，为家人写副对联呢？

—— 完 ——