神经网络识别手写优化（三）

前言

本文是为了实现存储自己训练好的模型 结构和参数，以及加载训练好的模型进行预测。

代码

保存

    def save(self,filename):        """        模型保存        :param filename: 文件名         :return:         """        data ={ "sizes": self.sizes, #模型结构                "weights": [w.tolist() for w in self.weights], #tolist转换为列表类型                "biases": [b.tolist() for b in self.biases],                "cost": str(self.cost.__name__) #保存一下损失函数        }        f=open(filename,"w")        json.dump(data,f)        f.close()

加载

def load(filename):    """    加载模型    :param filename:     :return:     """    f=open(filename,"r")    data=json.load(f)    f.close()    cost=getattr(sys.modules[__name__],data["cost"]) #找对象    net=Network(data["sizes"],cost=cost)    net.weights=[np.array(w) for w in data["weights"]]    net.biases=[np.array(b) for b in data["biases"]]    return net