def decode()

in datasets.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

    def decode(self, samples, logname):
        H = self.H
        out_samples = self.samples_to_image(samples)
        n_examples = out_samples.shape[0]
        d2 = H.sample_grid_dim
        if d2 > n_examples:
            d2 = n_examples
        d1 = n_examples // d2
        tiled_image = tile_images(out_samples, d1=d1, d2=d2)
        imname = f'{H.desc}-samples-{logname}.png'
        out_path = os.path.join(H.model_dir, imname)
        imageio.imwrite(out_path, tiled_image)
        self.logprint(f'Saved samples in file {out_path}')