Skip to content

ramchandracheke/dsb2018_topcoders

 
 

Repository files navigation

DSB2018 [ods.ai] topcoders 1st place solution

Model weights/annotated data

You can download the whole package (7.5G) that contains training data, nn models weights, GBT models.

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

License

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%