A collection of small extensions to Keras (RBM, momentum schedule, ..)
Python
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.

README.md

A collection of small extensions to Keras

Extensions

  • Restricted Boltzmann Machine (RBM)
  • Gaussian-Bernoulli Restricted Boltzmann Machine (GB-RBM)
  • Momentum schedule
  • Logging helpers (simultaneous logging to console and log file)

Note that some of these extensions are very coupled to Keras' internals which change from time to time. Whenever these extensions break due to changes in Keras, either the extensions need to be updated to reflect the changes, or an older version of Keras should be used.