刚入门DeepLearning不久,前一段时间一直在学习cifar10的分类,突然最近要做一个多标签的任务,突然有点不知所措,不知从何下手了。于是查阅了一些资料,了解一下多分类任务与多标签分类任务的异同。
- 多分类任务:只有一个标签,但是标签有多种类别。
-多标签分类任务:一条数据可能有一个或者多个标签,比如一个病人的眼底检测报告,它可能被标记患有糖尿病、高血压多个标签。
多标签分类任务的特点:1.类别标的数量是不确定的;2.类别标签之间可能存在相互依赖关系。
-在建立模型的时候,多分类任务一般采用softmax作为输出层的激活函数,用categorical_crossentropy作为损失函数;多标签分类一般采用sigmoid作为输出层的激活函数,用binary_crossentropy作为损失函数。
分类问题名称 | 输出层使用激活函数 | 对应的损失函数 |
---|---|---|
二分类 | sigmoid | binary_crossentrppy |
多分类 | softmax | categorical_crossentropy |
多标签分类 | sigmoid | binary_crossentropy |
表来自:https://blog.csdn.net/weixin_33796177/article/details/88022703