本人在搭建CNN神经网络时,发现了两种可以输出某一层的方法。
第一种:
functor = tf.keras.backend.function([model.layers[0].input],
[model.layers[-1].output])
predict=functor(trainX )
在该段代码中,第一个参数是模型的输入部分,第二个参数是最后一层的输出(可自行选择哪一层),调用该函数时,传入你的数据trainX,最后predict返回的就是最后一层的输出。
第二种:
functor=tf.keras.models.Model(inputs=model.input,outputs=model.layers[-1].output)
predict= layer_model.predict(trainX)
在该段代码中, 和上面的比较类似,也是需要输入两个参数,不过使用的predict函数进行预测。