deep-learning - 用于二进制分类的最终线性层中的输出数量

Question

对于二元分类，我们可以选择具有 1 个输出的最终线性层并使用具有阈值的 sigmoid，或者使用具有 2 个输出的最终线性层并使用 softmax。一个比另一个有什么优势吗？

标签： deep-learningpytorch

Answer 1

如果您正在进行二元分类，我建议使用 sigmoid 的 1 个输出节点，如果您的问题是多类分类，我建议使用与 softmax 的标签数量一样多的节点。