Awesome Open Source
Awesome Open Source

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:

  • [x] Utility Functions:
    • [x] Image Handler
    • [x] Metrics (Accuracy)
    • [x] Similarities (Cosine Similarity)
  • [x] LSTM Controller and Memory Unit
  • [x] Batch Generators
  • [x] 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]

Alternatives To Ntm One Shot Tf
Select To Compare


Alternative Project Comparisons
Related Awesome Lists
Top Programming Languages
Top Projects

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Python (807,639
Learning (76,220
Machine Learning (37,132
Deep Learning (36,513
Tensorflow (22,344
Neural (16,727
Neural Network (15,505
Data Science (10,158
Lstm (5,531
One Shot Learning (80
Neural Turing Machines (18