sbi-dev / pyknos

Conditional density estimation with neural networks
Apache License 2.0
27 stars 5 forks source link

Revamp model specification and parameterization #8

Open alvorithm opened 4 years ago

alvorithm commented 4 years ago

get_models from nflows doesn't scale, is annoying

Instead, as a minimum, we want a pluggable system (=extensible) for defining models that is based on templating.

For extra points:

What we don't want, under ANY circumstance: magic numbers sprinkled in the code, long command lines that disappear in the console's log (at best). What we want: declarative configuration with a measure of imperative flexibility (this is exactly what gin/thinc.config provide, see thinc's intro notebook).

The system should eventually be reusable in sbi.

alvorithm commented 4 years ago

A good practice test would be to create the models used in nsf for the uci dataset: linear_transform, base_transform and transform