Tensorflow Implementation of Relation Networks for the bAbI QA Task, detailed in "A Simple Neural Network Module for Relational Reasoning," [https://arxiv.org/abs/1706.01427] by Santoro et. al.
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data Add data Jul 9, 2017
model Add test and validation results Jul 19, 2017
preprocessor Add train loop Jul 11, 2017
.gitignore Update gitignore Jul 9, 2017
README.md Initial commit Jul 9, 2017
__init__.py Add data Jul 9, 2017
run.py Add test and validation results Jul 19, 2017
test.txt Add test and validation results Jul 19, 2017
valid.txt Add test and validation results Jul 19, 2017

README.md

Relation Networks

Tensorflow Implementation of Relation Networks for the bAbI QA Task, detailed in "A Simple Neural Network Module for Relational Reasoning," by Santoro et. al.

Punchline

...

Results

TBA

Components

...

Repository Structure

Repository is structured in the following way:

+ model/ - Model Definition Code, including RN specific module implementation

+ preprocessor/ - Preprocessing Code to load and vectorize the bAbI Tasks (joint training)

+ data/ - Raw bAbI Task Data File (10K Version)

+ run.py - Core script for training and evaluating the Relation Networks on the bAbI Tasks.