欢迎来到李白AI作诗机 👋

一款由人工智能驱动、能够自动生成中文诗歌的工具。

长久以来，我们一直梦想让机器自主创作诗歌。当无数作家和编辑尚未动笔时，AI早已完成了数千篇文章。如今，这便是迈出的第一步……

🏠 主页

👍 成果

经过对近四万首唐诗的学习，生成了以下诗句：

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

这首诗写得颇有韵味，正是辛勤学习的结果。可以说，几乎把整个《全唐诗》的精华都学透了，才练就了如此强大的能力，这可不是一般人能做到的吧？
在这篇博客中，我将详细讲解其中的技术细节。如果还有未尽之处，大家可以通过微信找到我——那个头像很特别的人。话不多说，先奉上GitHub链接。这个作诗机器人我会持续维护下去。如果大家时间紧张，来不及细读，不妨给项目点个星或者直接Fork一下，这样每次有更新我推送时，你都能第一时间看到。毕竟TensorFlow虽然已经进入1.0时代，但API仍会不断变化。

让我们一起为这个项目加满星星，让更多人了解并使用我们的作诗机器人。未来我们还将加入更多酷炫的功能，比如押韵等。

📥 安装

git clone https://github.com/jinfagang/tensorflow_poems.git

🛠 使用

# 训练模型
python3 train.py
# 生成诗歌
python3 compose_poem.py

训练过程中，你可能会看到类似这样的输出：

📈 更新

2018年8月16日

我们正式宣布启动一项新计划：StrangeAI学院——一个专注于人工智能学习与前沿算法交流的平台！我们坚信，人工智能应当服务于人类生活，而非被大型公司所掌控。
在这里，你可以提前了解我们的项目：http://ai.loliloli.pro（strangeai.pro即将上线）。

2018年3月12日

tensorflow_poems又“复活”了！虽然许久未更新，但不知不觉间已收获上千颗星，看来大家对它依然很感兴趣。非常感谢各位的关注，不过我们绝不能就此止步，这也是我重新回归的原因。接下来，我想向大家分享一些最新的进展。首先，强烈推荐大家关注我精心打造的知乎专栏——《人工智能从入门到逆天杀神》以及每周一期的黑科技专栏《TrackTech》。我们不仅关注人工智能，也致力于探索区块链等前沿技术：

• 《人工智能从入门到逆天杀神》（知乎专栏）：https://zhuanlan.zhihu.com/ai-man
• 《每周一项目黑科技—TrackTech》（知乎专栏）：https://zhuanlan.zhihu.com/tracktech
  如果你想讨论人工智能，可以访问我们的网站（目前暂用）：http://ai.loliloli.pro（strangeai.pro即将上线），同时请订阅我们的微信公众号：奇异人工智能学院。

2017年11月8日

似乎距离上次更新这个仓库已经很久了。这段时间有不少朋友通过微信联系我，甚至还有一些行业大佬。其实当初这个项目只是个练手的小玩意儿，当时我的技术还不太成熟，给大家带来了一些困扰。现在，“李白”强势回归！本次更新带来了以下改进：

• 对数据预处理脚本进行了前所未有的简化，即便是小学生也能轻松理解。
• 训练只需运行train.py即可，数据和预训练模型均已准备妥当。
• 可以直接使用compose_poem.py生成诗歌，这次不会再出现死循环问题了。

2017年6月1日 可能是最后一次更新

我决定在有空的时候重构这个项目。古诗承载着我骨子里的文艺情怀，最近手头的事情有些杂乱，因此特此召集大家，有兴趣的朋友欢迎加入QQ群：

292889553

2017年3月22日 重磅更新：新增藏头诗功能

这次的小更新解决了以下几个问题：

• 训练完成后无法生成诗歌的问题——原来是陷入了无限循环，现已修复。
• 新增了美化打印功能，生成的古诗格式规范，可以直接接入第三方应用或其他平台。
• 终端禁用了TensorFlow默认的调试信息。

最重要的是：我们的作诗机器人（暂称“李白”）现在可以根据你指定的字来创作藏头诗啦！！
欢迎大家继续支持，还没点星的快点星吧！保持更新，永远开源！
下面来看看“李白”创作的藏头诗：

# 最近一直下雨，就作一首以“雨”字开头的吧
雨霁开门中，山听淮水流。
落花遍霜霰，金壶横河湟。
年年忽息世，径远谁论吟。
惊舟望秋月，应柳待晨围。
人处山霜月，萧萧广野虚。

# 李白人工智能作诗机器人的作者长得比较帅，以“帅”字开头做一首吧
帅主何幸化，自日兼春连。
命钱犯夕兴，职馀玄赏圣。
君有不知益，浮于但神衍。
（浓浓的怀才不遇之风...）

👊 它不仅能写古诗，还能模仿周杰伦创作歌词！！

这是2017年3月9日更新的功能——模仿周杰伦歌曲风格创作歌词。大家可以先感受一下它的作品：

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

不得不说，由于缺乏足够的训练文本，目前AI生成的歌词还略显生硬……嗯，倒是带点忧郁的风格，但这显然和周杰伦的曲风不太一样。不过没关系，毕竟我们现在只有112首歌作为训练素材。在此呼吁大家一起来整理中国歌手的语料文本！！！
如果你喜欢周杰伦的歌，可以把他的每一首歌按行保存，每句之间用空格分隔，然后统一发到我的邮箱[jinfagang19@163.com]。相信随着训练文本的不断增加，我们的歌词创作机器人一定会越来越厉害！当然，我也会及时将更新后的数据集上传到GitHub，大家可以点个星关注项目的进展。

👥 作者

👤 jinfagang

• 网站：http://jinfagang.github.io
• GitHub：@JinTian

👤 William Song

• 网站：http://williamzjc.gitee.io/morninglake/
• GitHub：@Freakwill
• Twitter：@WilliamPython

👤 Harvey Dam

• GitHub：@damtharvey

👤 KnowsCount

• 网站：http://docs.knowscount.cc/
• GitHub：@KnowsCount

🎉 表达支持

如果这个项目对你有所帮助，请点亮一颗🌟！

Give a 🌟 if this project helped you!

📝 许可协议

版权 © 2020

tensorflow_poems 快速上手指南

本指南帮助你快速部署并使用基于 TensorFlow 的中文古诗生成工具（李白 AI 作诗机器人）。

环境准备

在开始之前，请确保你的开发环境满足以下要求：

• 操作系统：Linux / macOS / Windows (需配置 Python 环境)
• Python 版本：Python 3.x
• 核心依赖：TensorFlow (建议版本 1.x，因项目原始代码基于 TF 1.0 开发，若使用 TF 2.x 可能需要调整代码或启用兼容模式)
• 其他依赖：git

提示：国内用户安装 TensorFlow 时，推荐使用清华或阿里镜像源加速：

pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

安装步骤

1. 克隆项目仓库 打开终端，执行以下命令下载源代码：

   git clone https://github.com/jinfagang/tensorflow_poems.git
   

2. 进入项目目录

   cd tensorflow_poems
   

3. 安装依赖库 如果项目包含 requirements.txt，请运行：

   pip install -r requirements.txt
   

   若无该文件，请确保已手动安装 tensorflow。

基本使用

本项目主要包含两个核心功能：模型训练与诗歌创作。

1. 训练模型

首次使用或想要重新训练模型时，运行以下脚本。程序将自动读取预置的唐诗数据进行训练。

python3 train.py

运行后你将看到训练过程中的损失值变化日志。

2. 创作诗歌

训练完成后（或使用已有模型），运行以下脚本即可让 AI 自动生成古诗：

python3 compose_poem.py

进阶功能：藏头诗

该项目支持指定首字创作（藏头诗）。你可以通过修改 compose_poem.py 中的输入参数或在运行时传入特定字符来实现。例如，若要生成以“雨”字开头的诗歌，需在代码中指定起始字或按照脚本提示输入。

注意：由于本项目基于较早期的 TensorFlow 版本开发，若在运行过程中遇到 API 兼容性报错，请检查 TensorFlow 版本或参考项目 Issue 区进行适配调整。