Skip to content

Latest commit

 

History

History
11 lines (8 loc) · 565 Bytes

README.md

File metadata and controls

11 lines (8 loc) · 565 Bytes

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.