Skip to content
 
 

Latest commit

 

History

242 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Usage

Dataset generation

run

python runner_datasets.py --configs configs/knn_clsf.json 

to generate datasets specified in config files in the config json.

kdtree.py

This only run the program fit once without prediction. We use it for debugging.

The scripts may take one argument that specify how many cores is used for low-level parallelization. Use

kdtree.py 112

to call the kd-tree knn with 112 cores.

About

scikit-learn_bench benchmarks various implementations of machine learning algorithms across data analytics frameworks. It currently support the scikit-learn, DAAL4PY, cuML, and XGBoost frameworks for commonly used machine learning algorithms.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages