relation-network icon indicating copy to clipboard operation
relation-network copied to clipboard

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.

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.