【Python项目】Python利用神经网络自动生成的“藏头诗”生成器 | 附带源码

相关文件

关注小编，私信小编领取哟！
当然别忘了一件三连哟~~

公众号：Python日志
可以关注小编公众号，会不定时的发布一下Python小技巧，还有很多资源可以免费领取哟！！
源码领取：加Python学习交流群：773162165 可以领取哟

开发工具

Python版本：3.7.8
相关模块：
pytorch模块；
pyqt5模块；
numpy模块；
pyttsx3模块；
以及一些python自带的模块。

环境搭建

安装Python并添加到环境变量，pip安装需要的相关模块即可。
想用GPU完成模型训练的读者请自行搜索相关的网络教程配置CUDA环境，相关文件中会提供小编训练好的模型，因此你也可以直接下载使用。

效果展示

视频效果

Python基于神经网络自动生成“藏头诗”项目！

图片效果

原理简介

这里我们用网上别人整理好的57580首唐诗作为训练数据集，先加载一下数据：

'''创建dataloader'''
def CreateDataloader(tang_resource_path, batch_size=16, num_workers=2):
    poems = np.load(tang_resource_path, allow_pickle=True)
    dataset = poems['data']
    index2word = poems['index2word'].item()
    word2index = poems['word2index'].item()
    dataset = torch.from_numpy(dataset)
    dataloader = torch.utils.data.DataLoader(
        dataset, batch_size=batch_size, shuffle=True, num_workers=num_workers,
    )
    return dataloader, index2word, word2index

然后用pytorch定义一个简单的生成模型：

'''诗歌生成模型'''
class Poem(nn