Skip to content

Implementation of mixture density network layer and objective function in Keras

Notifications You must be signed in to change notification settings

CellDynamics/keras-mdn

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

keras-mdn

Fitting data using neural network not with mean square error, but with a probablistic gaussian mixture model. This is an implementation of Christopher M Bishop's 1994 paper from http://eprints.aston.ac.uk/373/1/NCRG_94_004.pdf

  • mdn.py contains the implementation of the custom keras layer and objective function
  • main.py contains two test examples

Slightly different from the formula given in the paper, I have added a small epsilon term to the equation in (22) and (23) to avoid division by zero and taking log of zero

First test example is a 1d to 1d mapping

Alt text

Second test example is a 1d to 2d mapping Alt text

Thanks to http://blog.otoro.net/2015/11/24/mixture-density-networks-with-tensorflow/ for helpful guide on implementation

About

Implementation of mixture density network layer and objective function in Keras

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%