├── README.md ├── illustrations ├── funnel_mmd.png ├── ngf-vs-svgd.gif ├── svgd.gif └── svgd.png ├── nvgd ├── __init__.py ├── experiments │ ├── __init__.py │ ├── bayesian_logistic_regression.py │ ├── bnn.py │ ├── config.py │ ├── data │ │ ├── DATA.md │ │ └── covertype.mat │ ├── dataloader.py │ ├── funnel_dimension_sweep.py │ ├── funnel_mmd_comparison.py │ ├── nvgd_bnn.py │ ├── parse_results.py │ ├── sd_maxing.py │ ├── sgld_bnn.py │ ├── svgd_bnn.py │ ├── sweep_bnn.py │ └── sweep_nvgd_bnn.py └── src │ ├── __init__.py │ ├── distributions.py │ ├── flows.py │ ├── kernels.py │ ├── metrics.py │ ├── models.py │ ├── nets.py │ ├── plot.py │ ├── stein.py │ └── utils.py └── setup.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/README.md -------------------------------------------------------------------------------- /illustrations/funnel_mmd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/illustrations/funnel_mmd.png -------------------------------------------------------------------------------- /illustrations/ngf-vs-svgd.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/illustrations/ngf-vs-svgd.gif -------------------------------------------------------------------------------- /illustrations/svgd.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/illustrations/svgd.gif -------------------------------------------------------------------------------- /illustrations/svgd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/illustrations/svgd.png -------------------------------------------------------------------------------- /nvgd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nvgd/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nvgd/experiments/bayesian_logistic_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/bayesian_logistic_regression.py -------------------------------------------------------------------------------- /nvgd/experiments/bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/bnn.py -------------------------------------------------------------------------------- /nvgd/experiments/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/config.py -------------------------------------------------------------------------------- /nvgd/experiments/data/DATA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/data/DATA.md -------------------------------------------------------------------------------- /nvgd/experiments/data/covertype.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/data/covertype.mat -------------------------------------------------------------------------------- /nvgd/experiments/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/dataloader.py -------------------------------------------------------------------------------- /nvgd/experiments/funnel_dimension_sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/funnel_dimension_sweep.py -------------------------------------------------------------------------------- /nvgd/experiments/funnel_mmd_comparison.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/funnel_mmd_comparison.py -------------------------------------------------------------------------------- /nvgd/experiments/nvgd_bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/nvgd_bnn.py -------------------------------------------------------------------------------- /nvgd/experiments/parse_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/parse_results.py -------------------------------------------------------------------------------- /nvgd/experiments/sd_maxing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/sd_maxing.py -------------------------------------------------------------------------------- /nvgd/experiments/sgld_bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/sgld_bnn.py -------------------------------------------------------------------------------- /nvgd/experiments/svgd_bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/svgd_bnn.py -------------------------------------------------------------------------------- /nvgd/experiments/sweep_bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/sweep_bnn.py -------------------------------------------------------------------------------- /nvgd/experiments/sweep_nvgd_bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/experiments/sweep_nvgd_bnn.py -------------------------------------------------------------------------------- /nvgd/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nvgd/src/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/distributions.py -------------------------------------------------------------------------------- /nvgd/src/flows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/flows.py -------------------------------------------------------------------------------- /nvgd/src/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/kernels.py -------------------------------------------------------------------------------- /nvgd/src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/metrics.py -------------------------------------------------------------------------------- /nvgd/src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/models.py -------------------------------------------------------------------------------- /nvgd/src/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/nets.py -------------------------------------------------------------------------------- /nvgd/src/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/plot.py -------------------------------------------------------------------------------- /nvgd/src/stein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/stein.py -------------------------------------------------------------------------------- /nvgd/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/nvgd/src/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/langosco/neural-variational-gradient-descent/HEAD/setup.py --------------------------------------------------------------------------------