This package contains a Colaboratory notebook that loads a version of the dataset for the Paired associative inference task (length 3 and 4) presented in the ICLR 2020 submission (also on arXiv).
If you use the dataset, please cite:
@inproceedings{
banino2020memo:,
title={MEMO: A Deep Network for Flexible Combination of Episodic Memories},
author={Andrea Banino and Adrià Puigdomènech Badia and
Raphael Köster and Martin J. Chadwick and Vinicius Zambaldi and
Demis Hassabis and Caswell Barry and Matthew Botvinick and
Dharshan Kumaran and Charles Blundell},
booktitle={International Conference on Learning Representations},
year={2020},
url={https://openreview.net/forum?id=rJxlc0EtDr}
}