测试环境:
centos7 + python3.6 + pytorch0.4 +cuda9
下面是用模型生成的藏头诗(深度学习)
深宫昔时见,古貌多自有。
度日不相容,年年生一目。
学者若为霖,百姓贻忧厄。
习坎与天聪,优游宁敢屡。
训练数据
57580首诗歌,每首诗歌,书(pytorch入门与实践)的作者对其进行了预处理,每首诗歌长度125字符(不足补空格,超过则丢弃)
下面data.py文件用于提取数据
import numpy as np
import os
def get_data(conf):
'''
生成数据
:param conf: 配置选项,Config对象
:return: word2ix: 每个字符对应的索引id,如u'月'->100
:return: ix2word: 每个字符对应的索引id,如100->u'月'
:return: data: 每一行是一首诗对应的字的索引id
'''
if os.path.exists(conf.data_path):
datas = np.load(conf.data_path) #np数据文件
data = datas['data']
ix2word = datas['ix2word'].item()
word2ix = datas['word2ix'].item()
return data, word2ix, ix2word
配置文件
class Config(object):
""&#