query_completion

Finalización de consultas personalizada

Este repositorio contiene código para construir un LSTM LM para el cumplimiento de auto de consulta personalizado. El modelo junto con los resultados experimentales y las líneas de base se describen en nuestro documento ACL 2018, actualmente disponible en ARXIV. https://arxiv.org/pdf/1804.09661.pdf

Entrena un modelo usando ./trainer.py /path/to/expdir --data /path/to/data.tsv --valdata /path/to/valdata.tsv set Hyperparameters siguiendo el formato en default_params.json .

Descripción de los archivos de código:

• beam.py - código auxiliar para hacer la búsqueda de haz
• FactorCell.py - Implementación de la capa recurrente de FactorCell
• Model.py: define el gráfico TensorFlow para el modelo de idioma
• Trainer.py - Script para entrenar un nuevo modelo de Langauge
• Dynamic.py - Script para evaluar el modelo capacitado en nuevos usuarios