资源分类
》 资源 》算法》pytorch-poetry-gen

pytorch-poetry-gen

2019-10-11 | |  81 |   0 |   0

ytorch char-RNN

训练数据是全唐诗，网上英语的的char-RNN在诗词的效果上很一般，可能是因为中文诗词比较稀疏，加入一个embedding层之后训练会好很多。

没有将数据padding成固定格式，所以可能会比较慢

总的训练样本大概有50k首诗，最后效果还可以。

如果训练样本少于1000，比如我只拿李白的诗，会发现模型基本上将所有诗全记下了，最后生成的除了个别词不一样，都是样本里的东西。

所以感觉以前拿char-RNN训练汪峰歌词的多半是没拟合好，样本对此中文词来说也太稀疏了。

usage

train.py是训练模型用的，你可以修改dataHandler.py选择不同数据，如宋词

sample.py是生成语句的，修改代码执行：



sample.png   

result.png



pytorch

pytorch还是很好用的，看了Mxnet、Pytorch、Tensorflow三家对于RNN的实现，M和P的框架是不错的，实现比较方便，但M的代码有些乱。T的实现有点醉。

pytorch是以python优先的框架，所以代码看起来很舒服，组织得很好，值得学习。

诗词来源

https://github.com/jackeyGao/chinese-poetry

库中缺少cinese-poetry文件夹，直接上面这个库clone即可，里面还有宋词，不过没来得及玩



上一篇：Kind-PyTorch-Tutorial

下一篇：pytorch-REINFORCE

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...