Tensorflow Series 1: 使用LSTM实现古诗人工智能作诗

时隔多年终于实现了这个牛逼的作诗机器人。大家可以看看效果图感受一下

本文由清华大学较为牛逼的研究生金天同学原创,欢迎转载,但是请保留这段版权信息,如果你对文章有任何疑问,欢迎微信联系我:jintianiloveu。牛逼大神一一为你解答!

Tensorflow Series 1: 使用LSTM实现古诗人工智能作诗_第1张图片
Screen Shot 2017-03-09 at 3.37.11 PM.png

它已经不仅仅能够作古诗,还能模仿周杰伦创作歌词!!

这是2017-03-9更新的功能,模仿周杰伦歌曲创作歌词,大家先来感受一下它创作的歌词:

我的你的她
蛾眉脚的泪花
乱飞从慌乱
笛卡尔的悲伤
迟早在是石板上
荒废了晚上
夜你的她不是她
....

怎么说,目前由于缺乏训练文本,导致我们的AI做的歌词有点....额,还好啦,有那么一点忧郁之风,这个周杰伦完全不是一种风格呀。
然而没有关系,目前它训练的文本还太少,只有112首歌,在这里我来呼吁大家一起来整理中国歌手的语料文本!!!
如果你喜欢周杰伦的歌,可以把他的歌一首一行,每首歌句子空格分开保存到txt中,大家可以集中发到我的邮箱:
[email protected]
相信如果不断的加入训练文本我们的歌词创作机器人会越来越牛逼!当然我会及时把数据集更新到github上,大家可以star一下跟进本项目的更新。

阅遍了近4万首唐诗

龙舆迎池里,控列守龙猱。
几岁芳篁落,来和晚月中。
殊乘暮心处,麦光属激羁。
铁门通眼峡,高桂露沙连。
倘子门中望,何妨嶮锦楼。
择闻洛臣识,椒苑根觞吼。
柳翰天河酒,光方入胶明
Tensorflow Series 1: 使用LSTM实现古诗人工智能作诗_第2张图片
Screen Shot 2017-03-09 at 3.42.51 PM.png

这诗做的很有感觉啊,这都是勤奋的结果啊,基本上学习了全唐诗的所有精华才有了这么牛逼的能力,这一般人能做到?
本博客讲讲解一些里面实现的技术细节,如果有未尽之处,大家可以通过微信找到我,那个头像很神奇的男人。
闲话不多说,先把github链接放上来,这个作诗机器人我会一直维护,如果大家因为时间太紧没有时间看,可以给这个项目star一下或者fork,
我一推送更新你就能看到,主要是为了修复一些api问题,tensorflow虽然到了1.0,但是api还是会变化。
把星星加起来,让更多人可以看到我们创造这个作诗机器人,后期会加入更多牛逼掉渣天的功能,比如说押韵等等。

Install tensorflow_poems

  • 安装要求:
tensorflow 1.0
python3.5
all platform
  • 安装作诗机器人, 简单粗暴,一顿clone:
git clone https://github.com/jinfagang/tensorflow_poems.git

由于数据大小的原因我没有把数据放到repo里面,大家我的QQ: 1195889656 或者微信: jintianiloveu 我发给你们把,顺便给我们的项目点个赞哦!~

  • 使用方法:
# for poem train
python3 main.py -w poem --train
# for lyric train
python3 main.py -w lyric --train

# for generate poem
python3 main.py -w poem --no-train
# for generate lyric
python3 main.py -w lyric --no-train

  • 参数说明

-w or --write: 设置作诗还是创作歌词,poem表示诗,lyric表示歌词

--train: 训练标识位,首次运行请先train一下...

--no-train: 生成标识位

训练的时候有点慢,有GPU就更好啦,最后gen的时候你就可以看到我们牛逼掉渣天的诗啦!

这是它做的诗:

龙舆迎池里,控列守龙猱。
几岁芳篁落,来和晚月中。
殊乘暮心处,麦光属激羁。
铁门通眼峡,高桂露沙连。
倘子门中望,何妨嶮锦楼。
择闻洛臣识,椒苑根觞吼。
柳翰天河酒,光方入胶明

感觉有一种李白的豪放风度!

这是它作的歌词:

我的你的她
蛾眉脚的泪花
乱飞从慌乱
笛卡尔的悲伤
迟早在是石板上
荒废了晚上
夜你的她不是她
....

更多关于本项目

jinfagang.github.io

你可能感兴趣的:(Tensorflow Series 1: 使用LSTM实现古诗人工智能作诗)