/ntm-one-shot

One-shot Learning with Memory-Augmented Neural Networks

Primary LanguagePython

One-shot Learning with Memory-Augmented Neural Networks

Theano implementation of the paper One-shot Learning with Memory-Augmented Neural Networks. To try the classification example, download the Omniglot dataset and place it in the data/omniglot folder.

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