yongyehuang / DC-hi_guides

[Data Castle 算法竞赛] 精品旅行服务成单预测 final rank 11
http://www.dcjingsai.com/common/cmpt/%E7%B2%BE%E5%93%81%E6%97%85%E8%A1%8C%E6%9C%8D%E5%8A%A1%E6%88%90%E5%8D%95%E9%A2%84%E6%B5%8B_%E7%AB%9E%E8%B5%9B%E4%BF%A1%E6%81%AF.html
92 stars 56 forks source link
data-mining lightgbm stacking xgboost

HI GUIDES 精品旅行服务成单预测

比赛说明:精品旅行服务成单预测

文件结构

|- hi_guide
|  |- data        # 比赛提供的原始数据
|  |  |- test      |  |  |- trainingset       |  |- features      # 特征提取函数
|  |  |- action.py     # 行为特征
|  |  |- comment.py     # 评论特征
|  |  |- history.py     # 历史订单特征
|  |  |- profile.py     # 用户信息特征
|  |  |- train_data.csv   # 保存提取的训练特征  
|  |  |- test_data.csv   # 保存提取的测试集特征 
|  |- log        # 模型训练日志
|  |- result       # 模型预测结果
|  |- model       # 保存训练好的模型和特征重要度分析文件
|  |- data_helper.py  # 执行特征提取的代码
|  |- my_utils.py   # 工具函数库,主要用到其中的 xgb 特征重要度分析函数
|  |- m1_xgb.py   # xgb 模型
|  |- m2_lgb.py   # lgb 模型
|  |- m3_cgb.py   # catboost 模型
|  |- stacking.py   # stacking 模型融合
|  |- get_no_used_features.py   # 获取 xgb 和 lgb 中的特征重要度

使用方式

# run the single model
python -u m1_xgb.py
# run the stacking model
python -u stacking.py
# 注意事项
# train_data, test_data = load_feat(re_get=True, feature_path=feature_path)  # 如果没有修改特征,设置re_get=False,就会直接导入之前保存好的特征。