从文件加载的神经网络模型为什么不能预测任何负要素?(使用Torch / Lua语言)

我一直在使用Torch/Lua中的人工神经网络模型,能够学习如何识别用于特定生物项目的真实夫妻(1)和假夫妻(0)。

算法的行为方式如下:在优化阶段中,软件会选择最佳超参数组合来优化神经网络,而在训练阶段中,软件会使用所选的最佳超参数组合来训练神经网络,并最终将训练好的模型保存到文件中(如说明所述)。在训练期间,模型预测真实夫妻和假夫妻。

到目前为止一切运作良好。

之后,在测试阶段,我的软件加载先前保存的模型文件,并将其应用于新的数据集。

问题来了:在测试过程中,神经网络惊人地只预测出了1(真实夫妻),没有预测出0(假夫妻)。

我已经为这个问题工作了几周,但我真的无法理解为什么会发生这种情况......神经网络模型在训练期间正确地预测了1和0,那么为什么在测试中没有预测出任何0呢?也许在模型保存过程中发生了一些无法解释的事情。

有关正在发生的事情的任何想法?

任何建议都非常感激!

点赞