我如何测试CNN模型比python中的训练集更少的类



我有热带风暴的图像,并试图建立一个模型来对风暴类别进行分类。在这里,我试图使用以前的风暴图像来预测最后一个风暴阶段,但"最后一场风暴"只涉及5个类别。训练集有7个类别(基本上我将数据集分为上一场风暴进行测试和第一场风暴作为训练(。最后,我不得不问,有什么方法可以预测比训练类更少的类别吗。

在我看来,只要你关心的5个类别仍然可以由模型预测,那么你的测试集是否包含比训练类别更少的类别并不重要。当模型为给定的测试样本生成预测时,您可以对预测的类进行排序,只选择精度最高的预测类(例如,前3个精度(,而忽略其余类

否则,我建议只使用您关心的类的数量来训练您的模型(5(,只使用这些类的训练集。

最新更新