Skip to main content

Universal and customizable implementation of the Hierarchical Variational Autoencoder architecture.

Project description

hVAE - backbone

This repository contains:

  • customizable backbone implementation of hVAE
  • reuseable hVAE components such as blocks, layers, losses, etc.
  • training, evaluation and analyzation scripts
  • checkpoint handling (using Weights & Biases)

Installation

pip install hvae_backbone

Usage

This repository is intended to be used as a backend package.
Please refer to hvae template repository for usage instructions.

Project Structure

├── hvae_backbone
│   ├── elements
│   │   ├── __init__.py             
│   │   ├── data_preproc.py         # Modules for data preprocessing
│   │   ├── dataset.py              # Base dataset class
│   │   ├── distributions.py        # Distributions, distributions generation
│   │   ├── layers.py               # Layers for building models
│   │   ├── losses.py               # Loss functions
│   │   ├── nets.py                 # Network architectures
│   │   ├── optimizers.py           # Optimizers
│   │   ├── schedules.py            # Schedules e.g. LR, KL weight
│   ├── __init__.py                 # package level scripts
│   ├── analysis.py                 # Analysis tools for trained models
│   ├── block.py                    # Blocks for building hierarchical models
│   ├── checkpoint.py               # Checkpoint handling (save, load)
│   ├── functional.py               # Functional scripts (training, loss, etc.)
│   ├── hvae.py                     # General hVAE class
│   ├── sequence.py                 # General sequential hVAE class
│   ├── utils.py                    # Utility functions

TODO:

  • callbacks
  • preprocessing
  • sample vs rsmaple blokkoknál (SimpleGenBlock)
  • weight initialization

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

hvae_backbone-0.1.126.tar.gz (34.5 kB view details)

Uploaded Source

Built Distribution

hvae_backbone-0.1.126-py3-none-any.whl (43.1 kB view details)

Uploaded Python 3

File details

Details for the file hvae_backbone-0.1.126.tar.gz.

File metadata

  • Download URL: hvae_backbone-0.1.126.tar.gz
  • Upload date:
  • Size: 34.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.8.16

File hashes

Hashes for hvae_backbone-0.1.126.tar.gz
Algorithm Hash digest
SHA256 e7691dac295eb4aaaeb0d41fe461af145bac31e0aae070549cd7b77bdccdffa4
MD5 b7009ab439d485538cba273c6463b3f8
BLAKE2b-256 3327acf659493dc6ab1df92c82decf649d928794886b8c22b55d924172a092ae

See more details on using hashes here.

File details

Details for the file hvae_backbone-0.1.126-py3-none-any.whl.

File metadata

File hashes

Hashes for hvae_backbone-0.1.126-py3-none-any.whl
Algorithm Hash digest
SHA256 c912e13a4d9214a05b937b860cafad0d07b9286b7bf8ed9efa683120a412a388
MD5 6ed91d9855ff9927fd339180f045236a
BLAKE2b-256 2f736cafcbbeda700bb9b079ab88091f0574a2c1b14ea401e9a2122fb9385787

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page