search

keras-team / keras-applications

Reference implementations of popular deep learning models.
Other
2k stars 913 forks source link

Keras decoder.predict won't work #137

Closed Elesa closed 4 years ago

Elesa commented 4 years ago

VAE_test.txt

Hello,

This is just to let you know that the keras VAE example (https://keras.io/examples/variational_autoencoder/)

produces error when you try to employ "decoder.predict" (if you run x_encoded = encoder.predict(x_test) x_decoded = decoder.predict(x_encoded) )

Thanks, Maria

taehoonlee commented 4 years ago

@Elesa, Please refer to the decoding parts. The correct usage is as follows:

z_mean, _, _ = encoder.predict(x_test)
x_decoded = decoder.predict(z_mean)
taehoonlee commented 4 years ago

@Elesa, I'll close the issue for now. Please feel free to open it again at any time if you have additional comments.