Keras, How to get the output of each layer?

后端 未结 10 1855
借酒劲吻你
借酒劲吻你 2020-11-22 07:34

I have trained a binary classification model with CNN, and here is my code

model = Sequential()
model.add(Convolution2D(nb_filters, kernel_size[0], kernel_si         


        
10条回答
  •  失恋的感觉
    2020-11-22 07:54

    Based on all the good answers of this thread, I wrote a library to fetch the output of each layer. It abstracts all the complexity and has been designed to be as user-friendly as possible:

    https://github.com/philipperemy/keract

    It handles almost all the edge cases

    Hope it helps!

提交回复
热议问题