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

This project contains a tensorflow 1.9.0 and a tensorflow 2.3.0 implementation of the Tolman-Eichenbaum Machine (paper). The tensorflow 2 (tem_tf2) version is cleaner and easier to use.

In the tensorflow 2 (tem_tf2) version, it has all the relevant code for the simulations in the recent Nature Neuroscience review (paper).

Getting Started

You need to install python 3 and tensorflow 1.9.0 or tensorflow 2.3.0

Installation

Clone the repo

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

Running models

python3 run_tem.py

If using graph_mode parameter, there will be some time before training starts as graph optimisaiton is taking place.

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!