• Stars
    star
    202
  • Rank 193,691 (Top 4 %)
  • Language
    Jupyter Notebook
  • Created over 6 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

唐诗,藏头诗,按需自动生成古诗,基于Keras、LSTM-RNN。文档齐全。

Poems_generator_Keras

 

  • 唐诗,古诗,五言绝句自动生成,基于Keras,LSTM-RNN。      
  • 附带训练好的模型文件,可直接上手使用。        
  • 功能:藏头诗,随机写诗,给定第一句诗/字进行作诗          

测试结果:

在GPU Tesla K80上,2s/epoch,一共有3w+个epoch。训练时的测试结果令人满意:

==================Epoch 4304=====================
县幽公事稀,上仙晓更高。风行随时朝,还云避倚里。
病客与僧闲,来王不鹤星。火气北所晚,边飞无已去。
玉律阳和变,下石凤明君。对动晨桂步,飞群安行金。
==================Epoch 4308=====================
绮阁云霞满,地国五自去。云人芳国思,云堂兵曲中。
帝城深处寺,此梦与云色。朝枝使天何,水天开光时。
石门千仞断,乡金在画使。天林东去结,北里石叶锦。
==================Epoch 4312=====================
王室今如毁,汉云树来寒。路平应人江,山不开云古。
扰扰走人寰,文尘李气奉。时秋田客岁,高斗不南中。
何以保孤危,都书丹道边。惜言为日芳,波垂日桃花。
==================Epoch 4316=====================
爱酒如偷蜜,心若大如去。入忆似烟春,如成台忆圣。
我爱窦高士,如箭变素外。明成丹泥为,今风酒影重。
画楼吹笛妓,何从出还玉。遍阴火川下,乘合未云虚。

环境配置

  • python3
  • tensorflow
  • Keras
  • h5py
  • Jupyter
  • numpy……

  食用指南

from config import Config
#加载模型(若无训练好的模型,会开始训练)
model = PoetryModel(Config)
print('model loaded')

#藏头诗
sen = model.predict_hide('争云日夏')
print(sen)

  输出结果:

争空谁上尽,云云中林翠。日落危西烟,夏更无长塞。

  其他方法调用请看notebook,里面都有。

如何使用训练好的模型:

  1. 我训练好了一个h5模型,点击页面中右侧的小箭头可下载,迅雷满速
  2. poetry_model.h5放入根目录,记得改一下self.loaded_model = True
  3. 注意,此模型只跑了4000+epoch,还有提升空间    

更新说明:

Version 1.1:

  • 代码大幅度重构,更加简洁
  • 添加多个模型方法,可按需要生成诗句
  • notebook中添加样例

Version 1.0:
    在ioiogoo的架构之上进行优化:

  • 改用Jupyter notebook
  • 数据只使用五言绝句,否则输出不稳定
  • 修复bugs
  • 精简代码
  • 训练时的测试数据会写入out/out.txt
  • 训练时的测试输入为随机一首诗的开头,确保输出好看
  • ………………

More Repositories

1

PI-REC

🔥 PI-REC: Progressive Image Reconstruction Network With Edge and Color Domain. 🔥 图像翻译,条件GAN,AI绘画
Python
2,028
star
2

Anime-InPainting

An application tool of edge-connect, which can do anime inpainting and drawing. 动漫人物图片自动修复,去马赛克,填补,去瑕疵
Python
1,027
star
3

Mask_Danmu

基于YOLOv2 / Mask-RCNN实现的视频蒙版弹幕,达到bilibili的demo效果。
Python
88
star
4

sepconv_video

基于SpeConv深度学习的视频补帧 插帧
Python
48
star
5

Identify_cat

Logistic Regression with a Neural Network mindset to identify cat photos.
Jupyter Notebook
25
star
6

MVPTest_login

非常简洁的mvp架构教程,关于一个简单的登陆系统,注释写的很详细
Java
21
star
7

Cosine_Stateful_Lstm

有关keras中stateful LSTM模型讲解的配套代码
Jupyter Notebook
18
star
8

YouYuMall

基于MVP架构的完整电商android应用开发(包含通用架构设计)
Java
13
star
9

MaterialTest

印象南京App,mvp+retrofit+rxjava,包含登陆,上传,服务器读取功能
Java
5
star
10

LCY_OnlineJudge

NJU_2018高级算法课程OJ答案
Python
4
star
11

kNN_Practice

kNN (k-邻近算法)实现与实战案例
Python
3
star
12

youyuge34.github.io

个人域名博客 Personal Blog Site used hexo
HTML
3
star
13

ACM_Personal_Training

个人刷的一些算法题,包括PAT甲乙级(牛客网),带数字序号的是九度oj的考研机试题 http://ac.jobdu.com/
JavaScript
3
star
14

VoteSite

一个基本的投票应用。它包含两部分: 一个公开的网站,可以让访客查看投票的结果并让他们进行投票。 一个后台管理网站,你可以添加、修改和删除选票。
Python
2
star
15

logRegres

Logistic回归分类器,最优化理论,梯度下降最优化算法,疝气病预测马匹死亡的实例
Python
2
star
16

DecisionTree

构造ID3算法决策树,用matplotlib绘制决策树
Python
2
star
17

Jobbole_Spider

使用scrapy对伯乐在线,知乎,拉勾网进行爬虫爬取
Python
2
star
18

SVMTest

支持向量机,简化版smo,完整版platt smo函数,核函数,手写识别问题实例,注释充分
Python
1
star
19

bayes

朴素贝叶斯分类器,文本分类
Python
1
star
20

DarkerFlow

在DarkFlow基础上优化后的yolov2目标检测系统
Jupyter Notebook
1
star
21

LinearRegression

Predict the house price from the imooc course with Jupyter,sklearn.
Jupyter Notebook
1
star
22

TensorFlowDocument_Jupyter

Chinese TensorFlow Document with Jupyter and python2.7
Jupyter Notebook
1
star
23

DBJ_Infomation_System

python2+tornado+MySQL 简单的一个订单采购系统,类似购物车,原代码框架@吴凡,本人仅仅修改
HTML
1
star
24

Json2CSV

爬取豆瓣电影短评,并转换为csv格式文件
Jupyter Notebook
1
star
25

MVP_simplestTest

http://www.jianshu.com/p/5c133a8a2b0d 博文对应的最简单的MVP架构,注释写的比较详细
Java
1
star
26

Noir_Art

Share your design work and f**k the life together!
JavaScript
1
star
27

spider_imooc

简单的python2爬虫框架实现,实现简单的调度器、URL池、下载器、解析器、输出模块
HTML
1
star
28

ImoocXianYu

《漫尤——动漫资讯综合App》,注释丰富 特点是封装了视频自动播放(mediaPlayer+textureView)的sdk, 并且在组件封装上下了很大功夫。
Java
1
star