欢迎光临 x-algo
关注算法在工业界应用
Hi, 这是一个关注大数据算法在工业界应用的网站

Chinese Poetry Generation with Planning based Neural Network

本文生成诗歌的时候,从第一句开始都是通过模型生成,不是从已有的语料中挑选/组合得到,文章Chinese Poetry Generation with Recurrent Neural Networks是通过phrase组合和平仄约束得到第一句.

本文做法是通过一个规划模型(Planning model)进行主题的规划,通过这个主题模型确定每一句的主题,进行诗歌生成.

整体结构

13.2.65

Planning

用户提供的词太多就用textrank提取关键词,每一个词看做是一个顶点,左右共现边权+1,下面公式 \(w_{j,i}\) 表示的是边权重, \(S(V_i)\) 表示第i个顶点的最终得分,初始得分为1.0.d取值0.85.

13.2.66

用户提供的词太少,就需要扩展,这里用两种方法:

  1. 第一种就是通过RNN预测下一个主题词,训练的时候通过TextRank得到所有词的权重,然后从诗集中每一句提取出来权重最高的一个词,每一首诗都可以提取出来一个关键词序列,用这个序列训练一个RNN,然后用这个RNN进行预测.
  2. 从开放语料中扩展关键词,就是讲doc中和keyword距离较近词性为名词和形容词并且在诗歌的语料中出现过.

Generation

结构如下图:

13.2.67

  1. 数据一行一行生成,生成第一行的时候,只有 \(h_0\) 作为context信息
  2. 生成第二行的时候,第一行的信息和第二行的关键词作为上下文信息

实现

https://github.com/DevinZ1993/Chinese-Poetry-Generation

https://github.com/Disiok/poetry-seq2seq

未经允许不得转载:大数据算法 » Chinese Poetry Generation with Planning based Neural Network

分享到:更多 ()

评论 抢沙发

*

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址

关注大数据算法在工业界应用

本站的GitHub关于本站