/Black-Swan

KDD CUP 2017

Primary LanguagePython

Black-Swan

KDD CUP 2017 task2

初赛第一,复赛第四

环境要求

Python2.7

  • sklearn
  • statusmodel
  • pandas
  • numpy
  • XGBoost
  • LightGBM

R

  • data.table
  • dplyr
  • reshape2
  • lubridate
  • h2o

运行

  • cd src/main/
  • python run.py
  • cd NN
  • sh ./run.sh
  • python ensemble.py

说明

  • src/model1 ExtraTreeRegression
  • src/model2 Sarimax ,Linear Regression
  • src/model3 Random Forst
  • src/model4 XGboost,GBDT
  • NN Neural Network