TensorFlow2.0学习笔记(6)

该博客介绍了如何利用TensorFlow的tf.keras API创建一个名为IrisModel的自定义神经网络类。通过定义`__init__`方法设置网络结构,包括一个带softmax激活和L2正则化的全连接层,并在`call`方法中实现前向传播。此外,还展示了如何导入所需的Dense层和Model类,以及如何编译、训练和总结模型。
摘要由CSDN通过智能技术生成

用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

4.model.compile

5.model.fit

6.model.summary

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值