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))
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))