标签： cnn resnet softmax
我为3个课程训练了ResNet模型。
我想使用该模型预测另一个验证集。
但是当我使用预测函数时，它只显示与[0.]相同的结果。 1. 0.]用于包含3个类的另一个验证集。
我认为，如果我可以更改阈值，那么它将很好预测。
所以我只想知道softmax层之前的预测值。 （例如[0.1，0.8，0.1]）
我有一个训练有素的模型，如何在没有softmax层的情况下导入或加载模型？