我正在实现一个用于图像分类的 CNN 模型,我正在学习损失函数。有几种类型的损失函数来确定错误。但是,我们如何找出适合图像多分类模型的函数呢?
如何知道哪种损失函数适合图像分类?
数据挖掘
深度学习
美国有线电视新闻网
损失函数
2021-09-23 03:45:40
2个回答
如果你有两个类(即二元分类),你应该使用二元交叉熵损失。
如果你有两个以上,你应该使用分类交叉熵损失。
对于 keras 中的图像分类,具有多个类别,categorical_crossentropy
是一个好的开始。