Tensorflow implementation of Automatic Domain Feature Extraction and Personalized Integration (DFEI) framework.
Code for the paper: Large-Scale Multi-Domain Recommendation: an Automatic Domain Feature Extraction and Personalized Integration Framework.
python==3.6
tensorflow-gpu==1.15.0
python DFEI.py --domains [0,1,2,4,5,6] --use_domainid 0 --verbose 200 --batch_size 512 --file_folder ./data --embedding_dim 16 --layers [128,64,32] --keep_prob [0.9,0.8,0.7] --lr 1e-3 --decay 0.9
The instruction of commands has been clearly stated in the codes (see the parse_args function).
We use the public KuaiRand-1K dataset [https://github.com/chongminggao/KuaiRand].
Last Update Date: Apr. 12, 2024 (UTC+8)