/keras_extensions

A collection of small extensions to Keras (RBM, momentum schedule, ..)

Primary LanguagePythonMIT LicenseMIT

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.