如何知道哪种损失函数适合图像分类?

数据挖掘 深度学习 美国有线电视新闻网 损失函数
2021-09-23 03:45:40

我正在实现一个用于图像分类的 CNN 模型,我正在学习损失函数。有几种类型的损失函数来确定错误。但是,我们如何找出适合图像多分类模型的函数呢?

2个回答

如果你有两个类(即二元分类),你应该使用二元交叉熵损失。

如果你有两个以上,你应该使用分类交叉熵损失。

对于 keras 中的图像分类,具有多个类别,categorical_crossentropy是一个好的开始。