def analyse(self, data, runPath, epoch):
     zemb, zsl, kls_df = super(CUB_Image_Sentence_ft, self).analyse(data,
                                                                    K=10)
     labels = ['Prior', *[vae.modelName.lower() for vae in self.vaes]]
     plot_embeddings(zemb, zsl, labels,
                     '{}/emb_umap_{:03d}.png'.format(runPath, epoch))
     plot_kls_df(kls_df, '{}/kl_distance_{:03d}.png'.format(runPath, epoch))
Example #2
0
 def analyse(self, data, runPath, epoch):
     zemb, zsl, kls_df = super(MNIST, self).analyse(data, K=10)
     labels = ['Prior', self.modelName.lower()]
     plot_embeddings(zemb, zsl, labels, '{}/emb_umap_{:03d}.png'.format(runPath, epoch))
     plot_kls_df(kls_df, '{}/kl_distance_{:03d}.png'.format(runPath, epoch))