AI诗歌生成在线演示
该演示系统为利用Flask框架搭建的Web应用
功能为自动生成诗歌。
输入内容包括诗歌体裁、是否藏头、诗歌标题/藏头内容。
点击生成按钮会开始生成计时并禁用所有控件直到生成结束，生成时长一般不超过1s。
标题留空以随机生成。如果是藏头诗则藏头不可留空，并且为了使页面简洁藏头时随机生成标题。
模 型 介 绍
诗歌生成模型主要利用Pytorch库搭建，结构基于Transformer并参考了GPT。
与基于GRU相比，基于Transformer的模型训练进度更快，即经过相同时间的训练能得到更低的损失。
模型训练的大致流程为：给定长度为n的字符串s0:(n-1)，取s0:(n-2)进行k维词嵌入，得到的词向量矩阵Ak*(n-1)输入模型。模型输出未归一化的预测词向量矩阵Bk*(n-1)，与s1:(n-1)计算交叉熵损失。
训练数据来源于Github上的中文古诗收集项目chinese-poetry。