我希望将模型的概率输出用作另一个模型的特征。
例如,我想确定图片上的鸟类是什么,我想使用CNN,训练它然后将概率结果与其他数据一起使用,如鸟的大小和重量,并将其喂入svm。
我是否需要使用培训和测试集来使用CNN提取这些概率?我应该将我的数据集分成折叠,然后提取每个不同测试折叠的概率,还是我可以训练和测试我的所有数据并保存概率?
答案 0 :(得分:1)
测试集旨在验证分类器是否达到目标,或者设置超参数。在这种情况下,您对CNN的输出不感兴趣,因为它只是整个图片中的中间层。
话虽如此,您显然没有通过其输入反向传播SVM错误。这是两阶段模型的结果。如果您这样做,您将优化CNN以用作该特定SVM的输入。