训练写诗的AI模型

课程任务：基于古代唐诗宋词⽂本，做⼀个“会写诗”模型。属于NLP⽂本⽣成任务。 只给了3个数据集，没指定模型。

我直接在百度搜“ix2word.npy”文件名，果然搜到了作者源码嘿嘿。

借鉴：

深度学习-tensorflow2：03 - AI写诗_深度学习写诗_玩转测试开发的博客-CSDN博客

TensorFlow文本生成（AI 写诗）_tensorflow 生成文本_Steven·简谈的博客-CSDN博客

项目下载地址（包括了数据集）： GitHub - Stevengz/Poem_compose: 人工智能写诗

因为我们也不需要Git提交，所以直接下载zip压缩包然后解压就行了。

项目目录结构：

tang.zip：唐诗数据集
    data.npy
    ix2word.npy
    word2ix.npy
test.py：测试运行最终结果
train.py：训练主体程序
training_checkpoints：训练结果

用pycharm打开。

一、需要的环境

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras import losses
import numpy as np
import os

因为之前在sysrec环境中安装过TensorFlow了，所以直接切换到这个环境。 

二、训练

直接一键开始run

大概2.5个小时，训练完成后在项目文件夹下生成training_checkpoints文件夹，

每一次epoch后，都会保存一次模型文件，太爱了！ 

 

三、测试

也是一键run

 每次run出来的诗词都不一样。