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