用tensorflow API:tf.keras搭建网络八股(用类class搭建神经网络)
class可以写出带有跳连的非顺序网络结构(跳连就是上层输出可能是下几层部分的输入)
1.import
2.train,test
3.class MyModel(Model) model=MyModel
class IrisModel(Model):
def __init__(self):
super(IrisModel, self).__init__()
self.d1=Dense(3,activation='softmax',kernel_regularizer=tf.keras.regularizers.l2())
#定义网络结构快
def call(self, x):
y = self.d1(x)#调用网络结构快,实现前向传播
return y
model = IrisModel()
相比较上一篇代码,还要在最上面加上这一块
from tensorflow.keras.layers import Dense
from tensorflow.keras import Model