Skip to content

Commit c5774a7

Browse files
committed
add scikit-learn-intelex info
1 parent 327b386 commit c5774a7

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
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.

bench.py

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

198198
if not params.no_intel_optimized:
199199
try:
200-
from daal4py.sklearn import patch_sklearn
200+
from sklearnex import patch_sklearn
201201
patch_sklearn()
202202
except ImportError:
203-
print('Failed to import daal4py.sklearn.patch_sklearn.'
203+
print('Failed to import sklearnex.patch_sklearn.'
204204
'Use stock version scikit-learn', file=sys.stderr)
205205
params.device = 'None'
206206
else:

0 commit comments

Comments
 (0)