keras-io

/

drug-molecule-generation-with-VAE

like 8

Hello,

I was hoping to extract the encoder portion of the model, as described here
https://keras.io/examples/generative/molecule_generation/

But the model file uploaded only seems to contain the decoder

from huggingface_hub import from_pretrained_keras
model = from_pretrained_keras("keras-io/drug-molecule-generation-with-VAE")
model.summary()

Model: "decoder"
__________________________________________________________________________________________________
 Layer (type)                   Output Shape         Param #     Connected to                     
==================================================================================================
 input_6 (InputLayer)           [(None, 435)]        0           []                               

435 is indeed the size of the latent space used as input by the decoder

Are you planning to upload the encoder as well? It would be a huge help!