Variational Bayesian Bow tie Neural Networks with Shrinkage

This repository provides a Python implementation of the Variational Bow tie neural network (VBNN).

The VBNN is described and derived in a Variational Bayesian Bow tie Neural Networks with Shrinkage paper.

---

Repository has the following files:

• VBNN_class.py which contains the main code for the network.
• helpers.py which contains utilities for easy training and testing.
• test.ipynb notebook which shows how to run VBNN on a simulated example.