神经网络:对于二元分类,使用 1 或 2 个带有 VGG19 的输出神经元

问题描述 投票:0回答:1

我有两组图像(混凝土裂缝和未开裂的混凝土),所以它们是二元分类,我使用 vgg19 对它们进行分类。 当我使用 (1) 个神经元作为输出层并使用 softmax 时,我得到的精度为 0.5,并在 250 个 epoch 期间得到修复,而当我使用 2 个神经元和 softmax 时,精度提高到 0.9 以上。

那么,我是否必须使用 1 或 2 个神经元作为 VGG19 的二元分类输出?

deep-learning softmax vgg-net
1个回答
0
投票

对于二元分类,您应该使用 sigmoid 而不是 softmax。如果您要使用 softmax,那么您可能需要使用两个输出神经元,正如您已经尝试过的那样。

© www.soinside.com 2019 - 2024. All rights reserved.