query_completion

Personalized Query Completion

This repo contains code for building an LSTM LM for personalized query auto-completion. The model along with experimental results and baselines are described in our 2018 ACL paper, currently available on arXiv. https://arxiv.org/pdf/1804.09661.pdf

Train a model using ./trainer.py /path/to/expdir --data /path/to/data.tsv --valdata /path/to/valdata.tsv Set hyperparameters following the format in default_params.json.

Description of code files:

• beam.py - helper code for doing beam search
• factorcell.py - implementation of the FactorCell recurrent layer
• model.py - defines the Tensorflow graph for the language model
• trainer.py - script for training a new langauge model
• dynamic.py - script for evaluating trained model on new users