AI 智能写情诗、藏头诗模型训练

​ 一、AI 智能情诗、藏头诗展示

最近使用PyTorch的LSTM训练一个写情诗（七言）的模型，可以随机生成情诗、也可以生成藏头情诗。

​

在特殊的日子用AI生成一首这样的诗，是不是很酷！下面分享下AI 智能写情诗原理。

二、AI 智能写情诗原理

1、AI 智能写情诗思想

通过学习既有的情诗，随机或者依据给出的字，逐个完成下一个字的预测，直到一首诗全部预测完成。

预测过程大致如下：

​

2、情诗准备

在网上搜索并收集情诗（本文七言），如https://github.com/chinese-poetry/chinese-poetry

本文的情诗包含两部分：

第一部分：原始的七言情诗

原始七言情诗，如：

相见时难别亦难，
引用
东风无力百花残。
引用
春蚕到死丝方尽，
引用
蜡炬成灰泪始干。

第二部分：改造的七言情诗

为了把写的比较好的五言、词也纳入到AI的学习中，将其改造生成七言情诗：改造后的为两行一句，通过随机匹配两句组成一首诗，如：

日看镜中问相思，溯游宛在水中央。
雁字多相思几许，恰豆蔻词工梦好。

3、模型训练

3.1、处理情诗数据

第一步需要将准备好的情诗数据转换为PyTorch LSTM模型需要的词向量格式：
构建x，y

x=[相,见,时,难,别,亦,难,，,东,风,无,力,百,花,残,。,春,蚕,到,死,丝,方,尽,，,蜡,炬,成,灰,泪,始,干]
y=[见,时,难,别,亦,难,，,东,风,无,力,百,花,残,。,春,蚕,到,死,丝,方,尽,，,蜡,炬,成,灰,泪,始,干，。]

将x对应的字，对应词库索引，对应向量；将y对应的字，对应词库索引，大致的过程如下：

​

3.2、构建情诗模型

第二步就是使用LSTM模型训练情诗模型，大致过程如下：

​

三、「AI智能古诗」体验

相关的AIGC（AI Generate Content）应用「AI智能古诗」小程序上线啦！✨「AI智能古诗」设计体验请查看《「AI智能古诗」体验设计》。