如何在Keras中保存编码的输出

Question

我想在带有Tensorflow后端的Keras自动编码器模型中的解码器阶段之前保存编码部分。

例如;

encoding_dim = 210
input_img = Input(shape=(5184,))

encoded = Dense(2592, activation='relu')(input_img)
encoded1 = Dense(encoding_dim, activation='relu')(encoded)

decoded = Dense(encoding_dim, activation='relu')(encoded1)
decoded = Dense(5184, activation='sigmoid')(decoded)

我想encoded1在自动编码器训练后另存为csv文件。假设输出Dense将为(nb_samples, output_dim)。

谢谢

Answer 1

尝试：

autoencoder = Model(input_img, decoded)
encoder = Model(input_img, encoded1)

autoencoder.compile(loss=my_loss, optimizer=my_optimizer)
autoencoder.fit(my_data, my_data, .. rest of fit params)

numpy.savetxt("encoded1.csv", encoder.predict(x), delimiter=",")

此外-我不知道您拥有哪种数据，但我建议您使用linear激活功能是最后一层和mse损失功能。