Function generate

fashion-mnist-vae/train.js:120–126

* Generate an image and preview it on the console. * * @param {tf.LayersModel} decoderModel Decoder portion of the VAE. * @param {number} latentDimSize Dimensionality of the latent space.

(decoderModel, latentDimSize)

 * @param {number} latentDimSize Dimensionality of the latent space.
 */
  async function generate(decoderModel, latentDimSize) {
  const targetZ = tf.zeros([latentDimSize]).expandDims();
  const generated = (decoderModel.predict(targetZ));

  await previewImage(generated.dataSync());
  tf.dispose([targetZ, generated]);
  }

  async function saveDecoder(savePath, decoderModel) {
  const decoderPath = path.join(savePath, 'decoder');