python 中利用 pickle 保存变量
Published:
在编写 python 项目时,有时候希望将模型训练的结果进行保存,以便下次进行使用,在这里利用 pickle 实现。
使用 pickle
pickle
这个包是 python 自带的,不需要另外再去安装。
具体使用如下:
import pickle
# 保存变量
def save_variable(v,filename):
f = open(filename, 'wb')
pickle.dump(v, f)
f.close()
return filename
# 读取变量
def load_variable(filename):
f = open(filename, 'rb')
r = pickle.load(f)
f.close()
return r
if __name__=='__main__':
c = [1, 2, 3, 4, 5, 6, 7]
filename = save_variable(c, 'test.data')
d = load_variavle(filename)
print(d==c)
需要注意这里的 file 必须要是以二进制的形式进行操作(写入或读取)。