[问题] 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