在编写seq2seq生成古诗的代码中,由于使用了训练好的字向量,需要读入程序中,并且还有将字转换为序号的字典也需要读入内存中,因此需要使用序列化与反序列化的方法保存读取这两个文件。
dict的序列化与反序列化
with open("word2int.txt", "wb") as f:
pickle.dump(word2int,f)
with open("word2int.txt", "rb+") as f:
d = pickle.load(f)
采用如上的代码,其中word2int定义为一个dict,使用了pickle模块。
np.array的序列化与反序列化
np.save('vocmat', matrix)
data = np.load('vocmat')