Skip to content

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

License

Notifications You must be signed in to change notification settings

jasonYoge/NTM-One-Shot-TF

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

47 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Written for Tensorflow v0.12. Yet to add 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 based on Neural Turing Machine architecture in Tensorflow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%