请问有实现训练的分类器的保存吗
Opened this issue · 2 comments
zcg-joker commented
如题~
zhulei227 commented
这部分功能没有做实现,可通过pickle保持模型参数,给个demo:
class Model(object):
def __init__(self):
self.w1#模型参数1
self.w2#模型参数2
def save(self,path):
params=dict()
params["w1"]=self.w1
params["w2"]=self.w2
with open(path, "wb") as f:
pickle.dump(params, f)
def load(self,path):
with open(path, "rb") as f:
params=pickle.load(f)
self.w1=params["w1"]
self.w2=params["w2"]
zcg-joker commented
好嘞~谢谢!!!