Dual Learning for Query Generation and Query Selection in Query Feeds Recommendation

Code for the paper Dual Learning for Query Generation and Query Selection in Query Feeds Recommendation.

Environment

Tensorflow 1.14.0

Keras 2.4.3

Dataset

Our proposed dataset for query generation and query selection is released in ./dataset/

Training and evaluating

To run DL-mt5, please use main.py in ./DL-mt5/

To run DL-QEBERT, please use main.py in ./DL-QEBERT/