Tensorflow 神经网络 二分类 多分类 输出层网络结构 与loss计算的from_logits设置

本文探讨了Tensorflow中处理二分类问题的两种网络结构方案,涉及最后一层是否使用sigmoid激活函数及from_logits参数的设置。通过实验比较,强调了正确设置from_logits对于网络训练稳定性及损失函数计算的重要性。此外,还讨论了one_hot编码目标输出的情况。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

针对二分类问题,最后一层的输出可以采用不同的设置方式,关键取决于目标值y_target的数据处理形式。

目标输出 (0 or 1)

如果y_target处理成(N, 1)格式,即布尔型格式(但本质应该还是整型)
最后一层的网络结构,采用1个神经元的全连接网络:

方案一

tf.keras.layers.Dense(1)  #最后层
model.compile(loss=tf.keras.losses.BinaryCrossentropy(from_logits=True),
              optimizer=tf.keras.optimizers.Adam(1e
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值