ccx06 / allennlp_ranking_guide

A brief tutorial on document ranking with allennlp.

Document Ranking with AllenNLP

Accompanying code for the AllenNLP guide post here.

For continued development and updates, take a look at allenrank.

Usage

Data

Run python scripts/data_split.py "https://github.com/microsoft/MIMICS/raw/master/data/MIMICS-ClickExplore.tsv" to automatically download the dataset to your /tmp/ directory.

Training

Docker

You can use the docker-compose file to start a docker container with the latest image of AllenNLP:

docker-compose run train

Manually

First, install the dependencies:

pip install -r requirements.txt

Then:

allennlp train experiments/mimics.jsonnet -s /tmp/your_output_dir