├── .gitignore ├── .gitmodules ├── README.md ├── config.py ├── environment.yml ├── imgs ├── 2d │ ├── 2uniforms_lgf-maf_300_epochs.png │ └── 2uniforms_maf_300_epochs.png └── images │ ├── lgf-realnvp_cifar10_samples.png │ ├── lgf-realnvp_fashion-mnist_samples.png │ ├── realnvp_cifar10_samples.png │ └── realnvp_fashion-mnist_samples.png ├── lgf ├── datasets │ ├── __init__.py │ ├── gaussian.py │ ├── image.py │ ├── loaders.py │ ├── supervised_dataset.py │ ├── two_d.py │ └── uci.py ├── experiment.py ├── models │ ├── __init__.py │ ├── components │ │ ├── bijections │ │ │ ├── __init__.py │ │ │ ├── acl.py │ │ │ ├── affine.py │ │ │ ├── batchnorm.py │ │ │ ├── bijection.py │ │ │ ├── bnaf.py │ │ │ ├── invconv.py │ │ │ ├── linear.py │ │ │ ├── made.py │ │ │ ├── math.py │ │ │ ├── nsf.py │ │ │ ├── reshaping.py │ │ │ └── sos.py │ │ ├── couplers.py │ │ ├── densities │ │ │ ├── __init__.py │ │ │ ├── concrete.py │ │ │ ├── density.py │ │ │ ├── elbo.py │ │ │ ├── exact.py │ │ │ ├── gaussian.py │ │ │ ├── split.py │ │ │ └── wrapper.py │ │ └── networks.py │ ├── factory.py │ └── schemas.py ├── trainer.py ├── visualizer.py └── writer.py ├── main.py ├── setup.py └── tests ├── __init__.py ├── test_bijection.py ├── test_density.py └── test_neural_nets.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/config.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/environment.yml -------------------------------------------------------------------------------- /imgs/2d/2uniforms_lgf-maf_300_epochs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/imgs/2d/2uniforms_lgf-maf_300_epochs.png -------------------------------------------------------------------------------- /imgs/2d/2uniforms_maf_300_epochs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/imgs/2d/2uniforms_maf_300_epochs.png -------------------------------------------------------------------------------- /imgs/images/lgf-realnvp_cifar10_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/imgs/images/lgf-realnvp_cifar10_samples.png -------------------------------------------------------------------------------- /imgs/images/lgf-realnvp_fashion-mnist_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/imgs/images/lgf-realnvp_fashion-mnist_samples.png -------------------------------------------------------------------------------- /imgs/images/realnvp_cifar10_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/imgs/images/realnvp_cifar10_samples.png -------------------------------------------------------------------------------- /imgs/images/realnvp_fashion-mnist_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/imgs/images/realnvp_fashion-mnist_samples.png -------------------------------------------------------------------------------- /lgf/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/__init__.py -------------------------------------------------------------------------------- /lgf/datasets/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/gaussian.py -------------------------------------------------------------------------------- /lgf/datasets/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/image.py -------------------------------------------------------------------------------- /lgf/datasets/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/loaders.py -------------------------------------------------------------------------------- /lgf/datasets/supervised_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/supervised_dataset.py -------------------------------------------------------------------------------- /lgf/datasets/two_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/two_d.py -------------------------------------------------------------------------------- /lgf/datasets/uci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/datasets/uci.py -------------------------------------------------------------------------------- /lgf/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/experiment.py -------------------------------------------------------------------------------- /lgf/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/__init__.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/__init__.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/acl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/acl.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/affine.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/batchnorm.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/bijection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/bijection.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/bnaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/bnaf.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/invconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/invconv.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/linear.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/made.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/math.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/nsf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/nsf.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/reshaping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/reshaping.py -------------------------------------------------------------------------------- /lgf/models/components/bijections/sos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/bijections/sos.py -------------------------------------------------------------------------------- /lgf/models/components/couplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/couplers.py -------------------------------------------------------------------------------- /lgf/models/components/densities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/__init__.py -------------------------------------------------------------------------------- /lgf/models/components/densities/concrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/concrete.py -------------------------------------------------------------------------------- /lgf/models/components/densities/density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/density.py -------------------------------------------------------------------------------- /lgf/models/components/densities/elbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/elbo.py -------------------------------------------------------------------------------- /lgf/models/components/densities/exact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/exact.py -------------------------------------------------------------------------------- /lgf/models/components/densities/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/gaussian.py -------------------------------------------------------------------------------- /lgf/models/components/densities/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/split.py -------------------------------------------------------------------------------- /lgf/models/components/densities/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/densities/wrapper.py -------------------------------------------------------------------------------- /lgf/models/components/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/components/networks.py -------------------------------------------------------------------------------- /lgf/models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/factory.py -------------------------------------------------------------------------------- /lgf/models/schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/models/schemas.py -------------------------------------------------------------------------------- /lgf/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/trainer.py -------------------------------------------------------------------------------- /lgf/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/visualizer.py -------------------------------------------------------------------------------- /lgf/writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/lgf/writer.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/main.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_bijection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/tests/test_bijection.py -------------------------------------------------------------------------------- /tests/test_density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/tests/test_density.py -------------------------------------------------------------------------------- /tests/test_neural_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/lgf/HEAD/tests/test_neural_nets.py --------------------------------------------------------------------------------