keras提取网络中间层输出、中间层特征

最新推荐文章于 2021-12-02 15:12:27 发布

小白不怕

最新推荐文章于 2021-12-02 15:12:27 发布

阅读量2.3k

点赞数 2

分类专栏： keras

本文链接：https://blog.csdn.net/qing_mm/article/details/106844062

版权

keras 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

最近想用自编码网络降维一下数据，用堆叠自编码器SAE降维之后提取中间层特征

model = Sequential()

model.add(Dense(1024, activation='relu', input_shape=(1000,)))
model.add(Dense(512, activation='relu'))
model.add(Dense(256, activation='relu'))
model.add(Dense(64, activation='relu'))#第四层
model.add(Dense(256, activation='relu'))
model.add(Dense(512, activation='relu'))
model.add(Dense(1024, activation='relu'))

model.compile(loss='mse', optimizer='RMSprop', metrics=['mse'])

#提取第四层的输出
layer_model = Model(inputs=X_model.input, outputs=X_model.layers[4].output)

x_train_Dense = layer_model.predict(Xtrain)#使用这个提取出的模型进行预测，得到中间输出特征

#之后就可以使用x_train_Dense这个提取到的特征进行其他操作

如有错误，欢迎指出