Code Monkey home page Code Monkey logo

peotry_generate's Introduction

基于循环神经网络(RNN)的古诗生成器

项目基于循环神经网络,使用tensorflow 1.4实现。训练数据为四万多首古诗,

训练完成后,能随机生成古诗。


使用说明

1.预处理数据

古诗文件不能够直接使用,需要生成词汇表,将古诗文件转换成向量。

python process_data.py

2.训练模型

使用1中生成的文件训练神经网络。

python train.py

3.生成古诗

当模型训练完成之后,可以尝试生成古诗。输入以下命令可以进行生成,至于质量嘛...emmmm,凑合吧

python eval.py

结果示例1:

树阴飞尽水三依,谩自为能厚景奇。 莫怪仙舟欲西望,楚人今此惜春风。

结果示例2:

岩外前苗点有泉,紫崖烟霭碧芊芊。 似僧月明秋更好,一踪颜事欲犹伤?

4.生成藏头诗

模型做的比较小,训练数据也不足,在生成藏头诗上的表现不是很好。往往要生成很多次生成一个满意的= =娱乐功能。

使用该功能要修改代码(等闲了我将它改成命令行功能,就不用改代码了),将eval.py中以下代码:

if __name__ == '__main__':
    # generate_acrostic(u'天空')
    generate_poem()

改成

if __name__ == '__main__':
    generate_acrostic(u'天空')
    # generate_poem()

然后运行即可。

python eval.py

结果示例:

天序曾柏乌倾鱼,空老桐歌尘翁红。

peotry_generate's People

Contributors

aaronjny avatar

Watchers

James Cloos avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.