Skip to content

Latest commit

 

History

8 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Dive into scikit-learn's HistGradientBoosting Classifier and Regressor

Running benchmarks

  1. Install anaconda

  2. Setup environment

conda env create -f environment.yml
conda activate 2019-pydata-nyc-hist
  1. Run benchmarks for each library

First run will download the HIGGS dataset which is 2.6 GB!

# This is the number of cores (no hyperthreading)
export OMP_NUM_THREADS=12
python bench.py sklearn
python bench.py catboost
python bench.py lightgbm
python bench.py xgboost

About

No description, website, or topics provided.

Resources

Stars

12 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages