在上一节,我们实现了一维神经网络框架的搭建。但是那种框架只适合一个输入特征的情况,并不能实现通用,本节将在一维框架的基础上实现通用框架的搭建,并且添加了模型保存和加载的代码。
通用模型各节点的实现代码如下,具体实现原理可看注释:
基础节点作为父类。
class Placeholder(Node):
"""
作为x,k,b,weights和bias这类需要赋初始值和更新值的类
"""
def __init__(self,name=None,is_trainable=True