多标签分类的预训练

Question

我必须为多标签分类预先训练一个模型。我正在使用cifar10数据集进行预训练，我想知道我是否必须使用'categorical_crossentrpy‘(softmax)或'binary_crossentropy’(sigmoid)进行预训练，因为在第一种情况下，我遇到了多分类问题

Answer 1

您应该使用softmax，因为它为您提供了每个类的概率，无论它们有多少。正如您所写的，Sigmoid与binary_crossentropy一起使用，并在二进制分类中使用(因此在名称中使用二进制)。我希望现在一切都清楚了。