Skip to content

One Shot Learning using Memory-Augmented Neural Networks (MANN) based on Neural Turing Machine architecture in Tensorflow

License

Notifications You must be signed in to change notification settings

hmishra2250/NTM-One-Shot-TF

Repository files navigation

One Shot Learning using Memory-Augmented Neural Networks in Tensorflow.

Update: added support for Tensorflow v1*.

Tensorflow implementation of the paper One-shot Learning with Memory-Augmented Neural Networks.

Current Progress of Implementation:

  • Utility Functions:
    • Image Handler
    • Metrics (Accuracy)
    • Similarities (Cosine Similarity)
  • LSTM Controller and Memory Unit
  • Batch Generators
  • Omniglot Tester Code
  • Unsupervised Feature Learning through Autoencoders
  • Cattle/New Born Recognition

The benchmark dataset is Omniglot dataset. All the datasets should be placed in the data/ folder.

Adam Santoro, Sergey Bartunov, Matthew Botvinick, Daan Wierstra, Timothy Lillicrap, One-shot Learning with Memory-Augmented Neural Networks, [arXiv]

About

One Shot Learning using Memory-Augmented Neural Networks (MANN) based on Neural Turing Machine architecture in Tensorflow

Topics

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages