LibCorner / TextKBQA

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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).

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

About

License:BSD 3-Clause "New" or "Revised" License


Languages

Language:Python 95.4%Language:Shell 4.6%