├── .gitignore ├── .gitmodules ├── CHANGELOG.md ├── README.md ├── cif ├── datasets │ ├── __init__.py │ ├── gaussian.py │ ├── image.py │ ├── loaders.py │ ├── supervised_dataset.py │ ├── tabular.py │ └── two_d.py ├── experiment.py ├── metrics.py ├── models │ ├── __init__.py │ ├── components │ │ ├── bijections │ │ │ ├── __init__.py │ │ │ ├── acl.py │ │ │ ├── actnorm.py │ │ │ ├── affine.py │ │ │ ├── batchnorm.py │ │ │ ├── bijection.py │ │ │ ├── bnaf.py │ │ │ ├── invconv.py │ │ │ ├── linear.py │ │ │ ├── made.py │ │ │ ├── math.py │ │ │ ├── nsf.py │ │ │ ├── ode.py │ │ │ ├── planar.py │ │ │ ├── resflow.py │ │ │ ├── reshaping.py │ │ │ └── sos.py │ │ ├── conditional_densities │ │ │ ├── __init__.py │ │ │ ├── bernoulli.py │ │ │ ├── concrete.py │ │ │ ├── conditional_density.py │ │ │ └── gaussian.py │ │ ├── couplers.py │ │ ├── densities │ │ │ ├── __init__.py │ │ │ ├── cif.py │ │ │ ├── density.py │ │ │ ├── flow.py │ │ │ ├── gaussian.py │ │ │ ├── marginal.py │ │ │ ├── split.py │ │ │ └── wrapper.py │ │ └── networks.py │ └── factory.py ├── trainer.py ├── visualizer.py └── writer.py ├── config ├── __init__.py ├── config.py ├── dsl.py ├── gaussian.py ├── images.py ├── schemas.py ├── tabular.py └── two_d.py ├── environment-lock.yml ├── environment.yml ├── main.py ├── setup.py └── tests ├── __init__.py ├── schemas ├── cif_realnvp_schema.json └── realnvp_schema.json ├── smoke_test.sh ├── test_bijection.py ├── test_config.py ├── test_density.py ├── test_neural_nets.py └── test_schemas.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/.gitmodules -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/README.md -------------------------------------------------------------------------------- /cif/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/__init__.py -------------------------------------------------------------------------------- /cif/datasets/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/gaussian.py -------------------------------------------------------------------------------- /cif/datasets/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/image.py -------------------------------------------------------------------------------- /cif/datasets/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/loaders.py -------------------------------------------------------------------------------- /cif/datasets/supervised_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/supervised_dataset.py -------------------------------------------------------------------------------- /cif/datasets/tabular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/tabular.py -------------------------------------------------------------------------------- /cif/datasets/two_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/datasets/two_d.py -------------------------------------------------------------------------------- /cif/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/experiment.py -------------------------------------------------------------------------------- /cif/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/metrics.py -------------------------------------------------------------------------------- /cif/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .factory import get_density 2 | -------------------------------------------------------------------------------- /cif/models/components/bijections/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/__init__.py -------------------------------------------------------------------------------- /cif/models/components/bijections/acl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/acl.py -------------------------------------------------------------------------------- /cif/models/components/bijections/actnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/actnorm.py -------------------------------------------------------------------------------- /cif/models/components/bijections/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/affine.py -------------------------------------------------------------------------------- /cif/models/components/bijections/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/batchnorm.py -------------------------------------------------------------------------------- /cif/models/components/bijections/bijection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/bijection.py -------------------------------------------------------------------------------- /cif/models/components/bijections/bnaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/bnaf.py -------------------------------------------------------------------------------- /cif/models/components/bijections/invconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/invconv.py -------------------------------------------------------------------------------- /cif/models/components/bijections/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/linear.py -------------------------------------------------------------------------------- /cif/models/components/bijections/made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/made.py -------------------------------------------------------------------------------- /cif/models/components/bijections/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/math.py -------------------------------------------------------------------------------- /cif/models/components/bijections/nsf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/nsf.py -------------------------------------------------------------------------------- /cif/models/components/bijections/ode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/ode.py -------------------------------------------------------------------------------- /cif/models/components/bijections/planar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/planar.py -------------------------------------------------------------------------------- /cif/models/components/bijections/resflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/resflow.py -------------------------------------------------------------------------------- /cif/models/components/bijections/reshaping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/reshaping.py -------------------------------------------------------------------------------- /cif/models/components/bijections/sos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/bijections/sos.py -------------------------------------------------------------------------------- /cif/models/components/conditional_densities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/conditional_densities/__init__.py -------------------------------------------------------------------------------- /cif/models/components/conditional_densities/bernoulli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/conditional_densities/bernoulli.py -------------------------------------------------------------------------------- /cif/models/components/conditional_densities/concrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/conditional_densities/concrete.py -------------------------------------------------------------------------------- /cif/models/components/conditional_densities/conditional_density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/conditional_densities/conditional_density.py -------------------------------------------------------------------------------- /cif/models/components/conditional_densities/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/conditional_densities/gaussian.py -------------------------------------------------------------------------------- /cif/models/components/couplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/couplers.py -------------------------------------------------------------------------------- /cif/models/components/densities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/__init__.py -------------------------------------------------------------------------------- /cif/models/components/densities/cif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/cif.py -------------------------------------------------------------------------------- /cif/models/components/densities/density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/density.py -------------------------------------------------------------------------------- /cif/models/components/densities/flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/flow.py -------------------------------------------------------------------------------- /cif/models/components/densities/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/gaussian.py -------------------------------------------------------------------------------- /cif/models/components/densities/marginal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/marginal.py -------------------------------------------------------------------------------- /cif/models/components/densities/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/split.py -------------------------------------------------------------------------------- /cif/models/components/densities/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/densities/wrapper.py -------------------------------------------------------------------------------- /cif/models/components/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/components/networks.py -------------------------------------------------------------------------------- /cif/models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/models/factory.py -------------------------------------------------------------------------------- /cif/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/trainer.py -------------------------------------------------------------------------------- /cif/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/visualizer.py -------------------------------------------------------------------------------- /cif/writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/cif/writer.py -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/config.py -------------------------------------------------------------------------------- /config/dsl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/dsl.py -------------------------------------------------------------------------------- /config/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/gaussian.py -------------------------------------------------------------------------------- /config/images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/images.py -------------------------------------------------------------------------------- /config/schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/schemas.py -------------------------------------------------------------------------------- /config/tabular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/tabular.py -------------------------------------------------------------------------------- /config/two_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/config/two_d.py -------------------------------------------------------------------------------- /environment-lock.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/environment-lock.yml -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/environment.yml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/main.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/schemas/cif_realnvp_schema.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/schemas/cif_realnvp_schema.json -------------------------------------------------------------------------------- /tests/schemas/realnvp_schema.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/schemas/realnvp_schema.json -------------------------------------------------------------------------------- /tests/smoke_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/smoke_test.sh -------------------------------------------------------------------------------- /tests/test_bijection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/test_bijection.py -------------------------------------------------------------------------------- /tests/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/test_config.py -------------------------------------------------------------------------------- /tests/test_density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/test_density.py -------------------------------------------------------------------------------- /tests/test_neural_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/test_neural_nets.py -------------------------------------------------------------------------------- /tests/test_schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrmcornish/cif/HEAD/tests/test_schemas.py --------------------------------------------------------------------------------