【发布时间】:2020-10-29 01:31:15
【问题描述】:
我设计了一个由 softmax 函数预测的具有大量输出的神经网络模型。但是,我想在不修改其他层的架构的情况下将所有输出分类为 5 个输出。该模型在第一种情况下表现良好,但是当我减少输出数量时,它会失去准确性并得到不好的泛化。我的问题是:即使只有 5 个输出,有没有一种方法可以让我的模型表现良好?例如:在输出层之前添加dropout层,使用其他激活函数等。
【问题讨论】:
-
这是一个社区特定的问题。 stackoverflow 仅用于编码问题。对于理论和概念问题,请参考 stack.ai 或 datascience stack。
标签: python neural-network pytorch