Skip to content

paulxiong/dsb2018_topcoders

 
 

Repository files navigation

DSB2018 [ods.ai] topcoders 1st place solution

See also solution description on Kaggle

You need to setup your environment first. Please install latest nvidia drivers, cuda 9 and cudnn 7. After that run setup_env.sh script

How to run predict

unzip test data into data_test folder and

./predict_test.sh

Submission files will be in predictions folder (submission_0.csv, submission_1.csv). Individual model predictions will be also in predictions folder.

How to run training

Before training please remove models from:

  • albu/weights
  • selim/nn_models
  • victor/nn_models
  • victor/lgbm_models

after it run:

./train_all.sh
./tune_all.sh
./predict_oof_trees.sh

We use two stage training because we want to tune models on the stage1 data released 11.04. Every script goes into every folder and runs scripts to train models.

About

DSB2018 [ods.ai] topcoders

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 94.2%
  • Cuda 2.2%
  • Shell 1.6%
  • C++ 1.5%
  • C 0.5%