zhulei227/ML_Notes

请问有实现训练的分类器的保存吗

Opened this issue · 2 comments

如题~

这部分功能没有做实现,可通过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"]

好嘞~谢谢!!!