No description, website, or topics provided.
Python Shell
Switch branches/tags
Nothing to show
Clone or download
Latest commit 248a5d5 Nov 25, 2017
Permalink
Failed to load latest commit information.
code
data_formatted
kb
model_outputs model outputs Apr 23, 2017
text_kb initial commit, sharing implementation Apr 23, 2017
vocab initial commit, sharing implementation Apr 23, 2017
LICENSE
README.md Update README.md Sep 12, 2017
_config.yml Set theme jekyll-theme-hacker Apr 23, 2017
config.sh adding the right data and pushing a config for running the saved model Sep 12, 2017
config_data.sh initial commit, sharing implementation Apr 23, 2017
get_data.sh adding the right data and pushing a config for running the saved model Sep 12, 2017
run.sh initial commit, sharing implementation Apr 23, 2017
run_data.sh initial commit, sharing implementation Apr 23, 2017
test_saved_model_config.sh adding the right data and pushing a config for running the saved model Sep 12, 2017

README.md

This repo contains the tensorflow implementation of the paper "Question Answering on Knowledge Bases and Text using Universal Schema and Memory Networks".

Dependencies

  • TensorFlow <= 0.12

Training

I have set up training with most default params on a very small dataset so that it is easier to get started. Just running the script should work.

/bin/bash run.sh ./config.sh

Data

The processed data (train/dev/test split) is stored in data_formatted/ directory. To download the KB files used for the project run,

sh get_data.sh

After downloading the data, you will have to change the appropriate entries in the config.sh file (kb_file and text_kb_file).

Pretrained embeddings

The embeddings used for initializing the network can be downloaded from here

Model outputs

We are also releasing the output predictions of our model for comparison. Find them in the model_outputs directory.

Trained Model

We are also sharing our pretrained model. Get it here. The following will load the model and get the answers from the dev set. Please change the config appropriately.

sh run.sh ./test_from_saved_model.sh