mihaibujanca / generalising-structural-knowledge

The Tolman-Eichenbaum Machine

· Report Bug · Request Feature ·

Table of Contents

• About the Project
• Getting Started
  • Installation
  • Running models
  • Pytorch version
• Contact
• Acknowledgements

About The Project

Tensorflow 1.9.0 code to reproduce model figures from https://papers.nips.cc/paper/8068-generalisation-of-structural-knowledge-in-the-hippocampal-entorhinal-system and https://www.biorxiv.org/content/10.1101/770495v2

Getting Started

You need to install python 3 and tensorflow 1.9.0

Installation

Clone the repo

git clone https://github.com/djcrw/generalising-structural-knowledge.git

Running models

python3 run_tem.py

Will take ~1hrs before training starts

Use notebook to load and visualise cell representations and to do behavioural analyses

Pytorch version

Jacob Bakermans has made a pytorch implementation of TEM found here https://github.com/jbakermans/torch_tem

Contact

James Whittington - @jcrwhittington - jcrwhittington at gmail.com

Project Link: https://github.com/djcrw/generalising-structural-knowledge

Acknowledgements

Thanks to Jacob Bakermans for the cover image!