Skip to content

Latest commit

 

History

History
20 lines (16 loc) · 1.25 KB

README.md

File metadata and controls

20 lines (16 loc) · 1.25 KB

Abstract

Build a statistical or deep neural model for Question Answering either utilizing both the provided Wikidata Knowledge Graph and Wikipedia Text Corpus or just a single knowledge source. Any existing methods/libraries could be utilized. In this project, I have implemented an End-to-End Memory Net using Keras. Later, trained the model for 100 epochs on Facebook's bAbI dataset which has 1000 questions for training and 1000 for test. Also, the dataset has an array of text passages ranging from Single Fact Evidence to Multihop Reasoning based corpus.

See the model performance in action: Video

Execution Instructions

  • Clone the repository
  • Install the project dependencies: "pip install -r requirements.txt"
  • In the "qa_model.py" file set the hyperparameters (lstm_size, epochs, batch size, model type) accordingly.
  • Run the model: "python qa_model.py"
  • Assess model performance using the test questions in Test Questions file

Available Models

  1. LSTM
  2. Bi-Directional LSTM
  3. GRU
  4. End-to-End Memory Net

Author

Akshay Joshi