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)
Built Distribution
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | e7691dac295eb4aaaeb0d41fe461af145bac31e0aae070549cd7b77bdccdffa4 |
|
MD5 | b7009ab439d485538cba273c6463b3f8 |
|
BLAKE2b-256 | 3327acf659493dc6ab1df92c82decf649d928794886b8c22b55d924172a092ae |
File details
Details for the file hvae_backbone-0.1.126-py3-none-any.whl
.
File metadata
- Download URL: hvae_backbone-0.1.126-py3-none-any.whl
- Upload date:
- Size: 43.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.8.16
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c912e13a4d9214a05b937b860cafad0d07b9286b7bf8ed9efa683120a412a388 |
|
MD5 | 6ed91d9855ff9927fd339180f045236a |
|
BLAKE2b-256 | 2f736cafcbbeda700bb9b079ab88091f0574a2c1b14ea401e9a2122fb9385787 |