query_completion

Achèvement des requêtes personnalisées

Ce dépôt contient du code pour construire un LSTM LM pour la complétion automatique de requête personnalisée. Le modèle ainsi que les résultats expérimentaux et les lignes de base sont décrits dans notre article ACL 2018, actuellement disponible sur ArXIV. https://arxiv.org/pdf/1804.09661.pdf

Train un modèle en utilisant ./trainer.py /path/to/expdir --data /path/to/data.tsv --valdata /path/to/valdata.tsv set hyperparameters suivant le format dans default_params.json .

Description des fichiers de code:

• Beam.py - code d'assistance pour faire la recherche de faisceau
• factorcell.py - mise en œuvre de la couche récurrente FactorCell
• Model.py - Définit le graphique TensorFlow pour le modèle de langue
• Trainer.py - Script pour la formation d'un nouveau modèle Langauge
• Dynamic.py - Script pour évaluer le modèle formé sur les nouveaux utilisateurs