本项目包括两个子项目:利用rnn进行MNIST数据集分类,利用lstm进行古诗生成
learn_rnn | Contents |
---|---|
img | 理解RNN的相关截图 |
main | 学习代码 |
MNSIT数据集分类 | 完整的使用RNN进行MNIST分类的项目 |
LSTM古诗生成 | 完整的的使用LSTM进行藏头诗生成的项目 |
main模块 | 学习代码 |
---|---|
main_demo01 | 引入 |
main_demo02 | 参数维度,输入特征维度,输出特征维度 |
MNSIT数据集分类 | Contents |
---|---|
img | 测试图片 |
module | 模型 |
test | 测试 |
train | 训练 |
LSTM古诗生成 | Contents |
---|---|
model | 模型 |
test | 测试,生成藏头诗 |
train | 训练 |
train_word_2_vector | 训练词向量 |
dataset | 自定义数据集 |
poetry_7.txt | 七言古诗训练集 |
LSTM藏头诗模型构建.pptx | LSTM模型构建流程,以及参数变化 使用microsoft ppt 打开,可以查看注释 wps ppt好像无法查看注释 |
RNN参数维度讲解:
https://www.bilibili.com/video/BV1dZ4y1g7DE?p=4&vd_source=168ed2a82d904bea741505ff76857c55
RNN MNIST数据集分类,特别感谢:小土堆
代码风格,请参考learn_pytorch
LSTM生成藏头诗,特别感谢: