请直接查看guide.pdf了解赛题,该项目是初赛第一名的模型。
- scikit-learn
- tqdm
- pandas
- numpy
- scipy
- tensorFlow=1.12.0 (其他版本≥1.4且不等于1.5或1.6)
- Linux Ubuntu 16.04, 128G内存(64G应该足够),一张显卡
mkdir data
cd data
#Download data from https://pan.baidu.com/s/1ASQMms_u70psRgW_KEyT2Q
#Password: burw
unzip algo.qq.com_641013010_testa.zip imps_log.zip user.zip
cd ..
python src/preprocess.py
python src/extract_feature.py
python src/convert_format.py
1)缺失值NA用0填充
2)将Word2Vec和DeepWalk得到的embedding拼接起来,并且掩盖到5%的广告
3)将需要用key-values的稠密特征正则化到[0,1]之间
mkdir submission
python train.py