[pyhton]神经网络保存与加载

第一种方法利用pickle方式

import pickle
#训练模型
model.fit(x,y)
s = pickle.dumps(model)
#写入文件
f = open("file_name.model",'wb+')
f.write(s)
f.close()

#加载模型
f2 = open("file_name.medel",'rb+')
s2 = f2.read()
model1 = pickle.load(s2)

第二种方式利用joblib

from sklearn.externals import joblib
model.fit(x,y)
joblib.dump(model,"file_name.m")

#读取模型
joblib.load("file_name.m")