Function vae

fashion-mnist-vae/model.js:155–169

* The combined encoder-decoder pipeline. * * @param {tf.Model} encoder * @param {tf.Model} decoder * * @returns {tf.Model} the vae.

(encoder, decoder)

 * @returns {tf.Model} the vae.
 */
function vae(encoder, decoder) {
  const inputs = encoder.inputs;
  const encoderOutputs = encoder.apply(inputs);
  const encoded = encoderOutputs[2];
  const decoderOutput = decoder.apply(encoded);
  const v = tf.model({
    inputs: inputs,
    outputs: [decoderOutput, ...encoderOutputs],
    name: 'vae_mlp',
  })

  // console.log('VAE Summary');
  // v.summary();
  return v;
}

/**
 * The custom loss function for VAE.