cyto_dl.models.vae package

Subpackages

• cyto_dl.models.vae.o2_spharm_vae package
  • Submodules
    • cyto_dl.models.vae.o2_spharm_vae.o2_spharm_encoder module
    • cyto_dl.models.vae.o2_spharm_vae.o2_spharm_vae module
• cyto_dl.models.vae.priors package
  • Submodules
    • cyto_dl.models.vae.priors.abstract_prior module
      • Prior
    • cyto_dl.models.vae.priors.gaussian module
      • DiagonalGaussianPrior
      • IsotropicGaussianPrior
      • compute_tc_penalty()
    • cyto_dl.models.vae.priors.identity_prior module
      • IdentityPrior
    • cyto_dl.models.vae.priors.joint_prior module
      • JointPrior

Submodules

• cyto_dl.models.vae.base_vae module
  • BaseVAE
    • BaseVAE.calculate_elbo()
    • BaseVAE.calculate_rcl()
    • BaseVAE.calculate_rcl_dict()
    • BaseVAE.decode()
    • BaseVAE.encode()
    • BaseVAE.forward()
    • BaseVAE.model_step()
    • BaseVAE.sample_z()
• cyto_dl.models.vae.image_canon_vae module
• cyto_dl.models.vae.image_encoder module
• cyto_dl.models.vae.image_vae module
• cyto_dl.models.vae.latent_loss_vae module
• cyto_dl.models.vae.point_cloud_vae module
• cyto_dl.models.vae.tabular_vae module