Skip to content

Commit 340a628

Browse files
author
Igor Rukhovich
committed
Merge remote-tracking branch 'origin/master' into xgb-nvidia-datasets
2 parents 6e47423 + 9a812bc commit 340a628

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ Create a suitable conda environment for each framework to test. Each item in the
4040
* [**scikit-learn**](sklearn_bench#how-to-create-conda-environment-for-benchmarking)
4141

4242
```bash
43-
conda create -n bench -c intel python=3.7 scikit-learn daal4py pandas
43+
conda create -n bench -c intel python=3.7 scikit-learn scikit-learn-intelex pandas
4444
```
4545

4646
* [**daal4py**](daal4py_bench#how-to-create-conda-environment-for-benchmarking)
@@ -67,7 +67,7 @@ Run `python runner.py --configs configs/config_example.json [--output-file resul
6767

6868
runner options:
6969
* ``configs`` : configuration files paths
70-
* ``no-intel-optimized`` : use no intel optimized version. Now avalible for scikit-learn benchmarks. Default is intel-optimized version.
70+
* ``no-intel-optimized`` : using Scikit-learn without Intel(R) Extension for Scikit-learn*. Now avalible for scikit-learn benchmarks. Default starts with using Intel(R) Extension for Scikit-learn*.
7171
* ``output-file``: output file name for result benchmarks. Default is `result.json`
7272
* ``report``: create an Excel report based on benchmarks results. Need library `openpyxl`.
7373
* ``dummy-run`` : run configuration parser and datasets generation without benchmarks running.

azure-pipelines.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ jobs:
66
- script: |
77
conda update -y -q conda
88
export FORCE_DAAL4PY_SKLEARN=yes
9-
conda create -q -y -n bench -c conda-forge python=3.7 pandas scikit-learn daal4py tqdm
9+
conda create -q -y -n bench -c conda-forge python=3.7 pandas scikit-learn scikit-learn-intelex tqdm
1010
displayName: Create Anaconda environment
1111
- script: |
1212
. /usr/share/miniconda/etc/profile.d/conda.sh
@@ -37,7 +37,7 @@ jobs:
3737
- script: |
3838
. /usr/share/miniconda/etc/profile.d/conda.sh
3939
conda activate bench
40-
python runner.py --configs configs/testing/daal4py.json
40+
python runner.py --configs configs/testing/daal4py.json --no-intel-optimized
4141
displayName: Run bench
4242
- job: Linux_XGBoost_and_daal4py
4343
pool:
@@ -50,7 +50,7 @@ jobs:
5050
- script: |
5151
. /usr/share/miniconda/etc/profile.d/conda.sh
5252
conda activate bench
53-
python runner.py --configs configs/testing/daal4py_xgboost.json
53+
python runner.py --configs configs/testing/daal4py_xgboost.json --no-intel-optimized
5454
displayName: Run bench
5555
- job: Pep8
5656
pool:

bench.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -198,10 +198,10 @@ def parse_args(parser, size=None, loop_types=(),
198198

199199
if not params.no_intel_optimized:
200200
try:
201-
from daal4py.sklearn import patch_sklearn
201+
from sklearnex import patch_sklearn
202202
patch_sklearn()
203203
except ImportError:
204-
logging.info('Failed to import daal4py.sklearn.patch_sklearn.'
204+
logging.info('Failed to import sklearnex.patch_sklearn.'
205205
'Use stock version scikit-learn', file=sys.stderr)
206206
params.device = 'None'
207207
else:

0 commit comments

Comments
 (0)