使用模型中的概率作为特征到另一个

时间:2017-08-04 11:43:59

标签: machine-learning deep-learning

我希望将模型的概率输出用作另一个模型的特征。

例如,我想确定图片上的鸟类是什么,我想使用CNN,训练它然后将概率结果与其他数据一起使用,如鸟的大小和重量,并将其喂入svm。

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

1 个答案:

答案 0 :(得分:1)

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

话虽如此,您显然没有通过其输入反向传播SVM错误。这是两阶段模型的结果。如果您这样做,您将优化CNN以用作该特定SVM的输入。