csv - 如何在 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=",")

此外 - 我不知道您有什么样的数据，但我建议您使用最后一层的线性激活和 mse 损失函数。