In [1]:

from sklearn_benchmarks.reporting.utils import ReportingInfo
from pathlib import Path
import pandas as pd
from sklearn_benchmarks.utils import default_run_dir
from sklearn_benchmarks.config import DEFAULT_NOTEBOOKS
pd.set_option('display.max_colwidth', None)
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)

sklearn_benchmarks reporting¶

In [2]:

run_dir = default_run_dir()
notebooks = DEFAULT_NOTEBOOKS
static = False

In [3]:

# Parameters
static = True
notebooks = [
    "scikit_learn_intelex_vs_scikit_learn",
    "onnx_vs_scikit_learn",
    "gradient_boosting",
]
run_dir = "./results/local/20220322T182254/"

In [4]:

run_dir = Path(run_dir)

In [5]:

reporting_info = ReportingInfo(run_dir=run_dir, notebooks=notebooks, static=static)
reporting_info.display()

Last run: Tuesday 22 March, 2022 at 18:22:55.

Notebooks¶

scikit-learn-intelex (Intel® oneAPI) vs. scikit-learn benchmarks

ONNX Runtime (Microsoft) vs. scikit-learn benchmarks

Gradient boosting: randomized HPO benchmarks

Benchmark environment information¶

System¶


python	3.8.12 \| packaged by conda-forge \| (default, Jan 30 2022, 23:42:07) [GCC 9.4.0]
executable	/usr/share/miniconda/envs/sklbench/bin/python
machine	Linux-5.11.0-1028-azure-x86_64-with-glibc2.10

Dependencies¶


pip	22.0.4
setuptools	60.10.0
sklearn	1.0.2
numpy	1.22.3
scipy	1.8.0
Cython	None
pandas	1.4.1
matplotlib	3.5.1
joblib	1.1.0
threadpoolctl	3.1.0

Threadpool¶

	user_api	internal_api	prefix	filepath	version	threading_layer	architecture	num_threads
0	blas	openblas	libopenblas	/usr/share/miniconda/envs/sklbench/lib/libopenblasp-r0.3.18.so	0.3.18	pthreads	SkylakeX	2
1	openmp	openmp	libgomp	/usr/share/miniconda/envs/sklbench/lib/libgomp.so.1.0.0	None	NaN	NaN	2

Cpu_count¶


cpu_count	2
physical_cpu_count	2

Time report¶

Total time elapsed: 1h 24m 37s

Hyperparameters match benchmarks¶

	time elapsed
estimator
sklearnex_PCA	2s
sklearnex_Ridge	6s
sklearnex_KMeans_short	7s
sklearn_PCA	12s
sklearn_KMeans_short	12s
sklearn_Ridge	15s
sklearnex_KNeighborsClassifier_brute_force	21s
sklearnex_LinearRegression	21s
sklearnex_LogisticRegression	28s
sklearn_TSNE	35s
sklearnex_TSNE	36s
sklearnex_KNeighborsClassifier_kd_tree	1m 2s
sklearn_LogisticRegression	1m 3s
sklearn_LinearRegression	1m 6s
sklearnex_KMeans_tall	1m 9s
sklearn_KMeans_tall	1m 32s
sklearn_HistGradientBoostingClassifier_best	3m 27s
sklearn_KNeighborsClassifier_kd_tree	4m 37s
sklearn_KNeighborsClassifier_brute_force	15m 8s

Hyperparameters optimization benchmarks¶

	time elapsed
estimator
lightgbm	10m 2s
xgboost	10m 7s
sklearn_HistGradientBoostingClassifier	10m 21s
catboost_lossguide	10m 32s
catboost_symmetric	11m 16s