tensorflow打印出模型中任意特征层(三)(tensorflow模型可视化)
1、前面二个内容中将了使用model.add()和class()搭建的网络,将其模型的任何一层的特征给打印出来,那么这次的内容将介绍如何使用tensorflow的高级API搭建的模型,将模型的特征打印出来。
如果你是使用这种API(如下图)搭建的API,那么前两次的内容不适用与本次API搭建的可视化。
x = Conv1D(filters=None,kernel_size=None,strides=None,padding=None)(input)
下面开始介绍本次可视化的步骤。大致是你需要将你训练后的模型保存下来,然后加载模型,选择要可视化的层,把你选择的层的特征给打印出来,下面开始仔细的介绍。
2、首先你需要保存你的模型,使用下列的代码保存模型,最后在model.fit()中调用。具体如下。
input = tf.keras.Input(shape=())#根据自己的数据集设定形状
x1 = Conv1D(#里面的参数自几填写)(input)
x2