[問題] Keras如何取出層與層間的輸出參數

作者: eecheng87 (EEcheng)   2019-09-22 16:28:05
大家好, 小弟目前用Keras訓練了一個簡單的MNIST模型
接著要用python實作由Keras做好的模型(每一層的功能自己寫)
但現在遇到的問題是我想要debug時沒辦法一層一層確認是
哪一層出問題, 只能由一開始輸入的28*28和最後輸出的10*1來
debug.
我想請問一下, 有大大知道有甚麼方法可以拿到每層的輸出嗎?
(如: 第一次convolution運算後的26*26 output frame)
PS:
1) 目前我是用layers.get_weights()拿訓練好的參數出來
2) 由test_data set算的正確率本來應該要97, 但把這些
data set 拿去套我自己實作的模型只剩不到1%, 應該
是實作有錯
3) 我指的拿出每層參數, 前提是可以有輸入一開始的28*28
input, 否則應該也不可能拿出中間的數字
4) 若無法拿出中間的結果, 那是否有比較建議的方式來debug
先在這裡謝謝大家!!
作者: sxy67230 (charlesgg)   2019-09-22 17:16:00
Intermediate_output = Model(input, output) intermediate_output.predict([X])

Links booklink

Contact Us: admin [ a t ] ucptt.com