Skip to content

Latest commit

 

History

History

pt

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
evaluation.py
evaluation.py
 
 
main.py
main.py
 
 
myio.py
myio.py
 
 

README.md

Similar Question Retrieval (pre-training part)

This directory contains the implementation of the neural method described in the paper. The method is used to calculate text similarity, for applications such as similar question retrieval in community-based QA forums.

The code in this directory trains an encoder-decoder network using the unsupervised corpus. The encoder is subsequently fine-tuned for retrieval task (code available at code/qa directory).

Data

The data used in this work is taken from the AskUbuntu 2014 dump. The processed data can be downloaded at this repo.

Dependencies

To run the code, you need the following extra packages installed:

  • PrettyTable (only for this project)
  • Scikit-Learn (only for this project)
  • Numpy and Theano (required in general for this repository)
Usage
  1. Clone the rcnn repo
  2. Use “export PYTHONPATH=/path/to/rcnn/code” to add the rcnn/code directory to Python library
  3. Run python main.py --help to see all running options

To specify Theano configs, run the code via THEANO_FLAGS='...' python main.py ... For instance, here is an example to run the model with default parameters:

THEANO_FLAGS='device=gpu,floatX=float32'        # use GPU and 32-bit float
python main.py --corpus path/to/corpus          # path to the corpus file
      --embeddings /path/to/vectors             # path to load word vectors
      --train path/to/train                     # path to training file
      --dev path/to/dev                         # path to dev file for retrieval task
      --test path/to/test                       # path to test file for retrieval task
      --heldout path/to/heldout                 # path to heldout file for calculating perplexity
      --dropout 0.1                             # dropout probability
      -d 400                                    # hidden dimension
      --model model.pkl.gz                      # save trained encoder to this file

The saved encoder will be fine-tuned in the retrieval model at code/qa.

The corpus, heldout question file, training/development/test files and the word vectors are available at the data repo.