query_completion

Conclusão personalizada de consulta

Este repositório contém código para criar um LSTM LM para conclusão automática de consulta personalizada. O modelo, juntamente com os resultados experimentais e as linhas de base, são descritos em nosso artigo da ACL de 2018, atualmente disponível no ARXIV. https://arxiv.org/pdf/1804.09661.pdf

Treine um modelo usando ./trainer.py /path/to/expdir --data /path/to/data.tsv --valdata /path/to/valdata.tsv Definir hiperparameters seguindo o formato em default_params.json .

Descrição dos arquivos de código:

• Beam.py - Código auxiliar para fazer pesquisa de feixe
• fatorcell.py - Implementação da camada recorrente do fatorCell
• model.py - define o gráfico de tensorflow para o modelo de idioma
• treinador.py - script para treinar um novo modelo Langauge
• Dynamic.py - Script para avaliar o modelo treinado em novos usuários