2017-08-04 38 views
1

我想將模型中的概率輸出用作另一個模型的特徵。例如,我想確定一張照片上的鳥是什麼樣的,我想使用CNN,訓練它,然後用其他數據的概率結果,比如鳥的大小和重量,然後餵它到一個svm。將模型中的概率用作特徵到另一個

我需要使用訓練和測試集來提取這些概率使用CNN嗎?我是否應該將數據集放入摺疊中,然後提取每個不同測試摺疊的概率,或者我可以僅訓練和測試所有數據並保存概率?

回答

1

測試集旨在驗證分類器是否達到其目標,或者設置超參數。在這種情況下,你對CNN的輸出不感興趣,因爲它只是大圖中的一箇中間層。

話雖如此,你顯然不通過其輸入反向傳播SVM錯誤。這是兩階段模型的結果。如果你這樣做,你會優化CNN作爲輸入到特定的SVM。

+0

因此,在這種情況下培訓和測試單個CNN是否可行?它不是過度的嗎? –

+0

@LucasRamos:讓我們暫時假定CNN確實是過度配合。問題是CNN輸出對於訓練數據是最佳的,但是對其他數據有泛化錯誤。你會用相同的數據訓練SVM嗎?如果是這樣,那麼SVM將受到其輸入的泛化誤差的影響。否則,SVM將否定這些 – MSalters

相關問題