Code accompanying the paper Recurrent Relational Networks for Complex Relational Reasoning https://arxiv.org/abs/1711.08028
Clone or download
Latest commit d3a5a27 May 17, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
docs review comments Nov 28, 2017
tasks no padding May 17, 2018
tests initial release Nov 21, 2017
.gitignore blog post Nov 24, 2017
README.md fix README Nov 21, 2017
message_passing.py initial release Nov 21, 2017
model.py initial release Nov 21, 2017
requirements.txt initial release Nov 21, 2017
trainer.py Add 'Deeply Learning the Messages in Message Passing Inference' basel… Dec 14, 2017
util.py initial release Nov 21, 2017

README.md

Recurrent Relational Networks for Complex Relational Reasoning

Contains the code to reproduce the experiments in the paper Recurrent Relational Networks for Complex Relational Reasoning

Sudoku

  • python tasks/sudoku/train.py will train a RRN with the hyper parameters from the paper and save the best model in the working directory
  • python tasks/sudoku/test.py will load the best model from the working directory and evaluate it on the test set

BaBi

  • python tasks/babi/train.py will train a RRN with the hyper parameters from the paper and save the best model in the working directory
  • python tasks/babi/test.py will load the best model from the working directory and evaluate it on the test set