本文最后更新于:2024年5月7日 下午
keras支持模型多输入多输出,本文记录多输出时loss、loss weight和metrics的设置方式。
模型输出
假设模型具有多个输出
- classify: 二维数组,分类softmax输出,需要配置交叉熵损失
- segmentation:与输入同尺寸map,sigmoid输出,需要配置二分类损失
- others:自定义其他输出,需要自定义损失
具体配置
model
- 变量均为模型中网络层
1 |
|
loss
1 |
|
loss weight
1 |
|
metrics
1 |
|
编译
1 |
|
文章链接:
https://www.zywvvd.com/notes/study/deep-learning/keras/keras-multi-metrics/keras-multi-metrics/
“觉得不错的话,给点打赏吧 ୧(๑•̀⌄•́๑)૭”

微信支付

支付宝支付
Keras 模型多输出 loss weight metrics 设置
https://www.zywvvd.com/notes/study/deep-learning/keras/keras-multi-metrics/keras-multi-metrics/