/relation-network

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.

Primary LanguagePython

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.