@@ -43,8 +43,7 @@ requested operation (e.g. call to ``.fit()`` / ``.predict()``
4343on the estimator class being used) is :ref: `supported on device/GPU <sklearn_algorithms_gpu >`, computations
4444will be performed on the device where the data lives, without involving any data transfers.
4545
46- Note that all of the inputs (e.g. ``X `` and ``y `` passed to ``.fit() `` methods) must be allocated on the same device for this to
47- work. If the requested operation is not supported on the device where the data lives, then it will either fall
46+ If the requested operation is not supported on the device where the data lives, then it will either fall
4847back to |sklearn |, or to an accelerated CPU version from the |sklearnex | when supported - these are controllable
4948through options ``allow_sklearn_after_onedal `` (default is ``True ``) and ``allow_fallback_to_host `` (default is
5049``False ``), respectively, which are accepted by ``config_context `` and ``set_config `` after
@@ -67,11 +66,6 @@ be transferred to host if it isn't already, and the computations will happen on
6766 Enable :ref: `verbose ` to see information about whether data transfers happen during an operation or not,
6867 whether an accelerated version from the extension is used, and where (CPU/device) the operation is executed.
6968
70- When passing array API inputs to methods such as ``.predict() `` of estimators with array API support, the output
71- will always be of the same class as the inputs, but be aware that array attributes of fitted models (e.g. ``coef_ ``
72- in a linear model) will not necessarily be of the same class as array API inputs passed to ``.fit() ``, even though
73- in many cases they are.
74-
7569.. warning ::
7670 If array API inputs are passed to an estimator's ``.fit() ``, subsequent data passed to methods such as
7771 ``.predict() `` or ``.score() `` of the fitted model **must reside on the same device ** - meaning: a model that
@@ -128,11 +122,6 @@ The following patched classes have support for array API inputs:
128122 next subsection.
129123
130124.. note ::
131- While full array API support is currently not implemented for all classes, |dpnp_array | inputs are supported
132- by all the classes that have :ref: `GPU support <oneapi_gpu >`. Note however that if array API support is not
133- enabled in |sklearn |, when passing these classes as inputs, data will be transferred to host and then back to
134- device instead of being used directly.
135-
136125 Result attributes of |sklearnex | classes which contain |sklearn | or |sklearnex | classes may not themselves be
137126 array API compliant. For example, ensemble algorithms contain decision tree estimators result objects which
138127 do not comply with the array API standard.
0 commit comments