query_completion

Персонализированное завершение запроса

Этот репо содержит код для создания LSTM LM для персонализированного автоматического завершения запроса. Модель наряду с экспериментальными результатами и базовыми показателями описаны в нашей статье ACL 2018 года, в настоящее время доступная на ARXIV. https://arxiv.org/pdf/1804.09661.pdf

Обучить модель с использованием ./trainer.py /path/to/expdir --data /path/to/data.tsv --valdata /path/to/valdata.tsv Установите гиперпараметры после формата в default_params.json .

Описание файлов кода:

• beam.py - вспомогательный код для поиска луча
• factorcell.py - реализация рецидивирующего уровня фактора,
• model.py - определяет график Tensorflow для языковой модели
• Trainer.py - сценарий для обучения новой модели Langauge
• Dynamic.py - Скрипт для оценки обученной модели на новых пользователях