2017-08-04 74 views
1

我想将模型中的概率输出用作另一个模型的特征。例如,我想确定一张照片上的鸟是什么样的,我想使用CNN,训练它,然后用其他数据的概率结果,比如鸟的大小和重量,然后喂它到一个svm。将模型中的概率用作特征到另一个

我需要使用训练和测试集来提取这些概率使用CNN吗?我是否应该将数据集放入折叠中,然后提取每个不同测试折叠的概率,或者我可以仅训练和测试所有数据并保存概率?

回答

1

测试集旨在验证分类器是否达到其目标,或者设置超参数。在这种情况下,你对CNN的输出不感兴趣,因为它只是大图中的一个中间层。

话虽如此,你显然不通过其输入反向传播SVM错误。这是两阶段模型的结果。如果你这样做,你会优化CNN作为输入到特定的SVM。

+0

因此,在这种情况下培训和测试单个CNN是否可行?它不是过度的吗? –

+0

@LucasRamos:让我们暂时假定CNN确实是过度配合。问题是CNN输出对于训练数据是最佳的,但是对其他数据有泛化错误。你会用相同的数据训练SVM吗?如果是这样,那么SVM将受到其输入的泛化误差的影响。否则,SVM将否定这些 – MSalters

相关问题