-
Notifications
You must be signed in to change notification settings - Fork 179
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[experiment] ENH: using only raw inputs for onedal backend #2153
base: main
Are you sure you want to change the base?
Conversation
not tested properly yet
sklearnex/cluster/dbscan.py
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Make sense for dbscan and rf use just onedal4py API only for raw inputs.
onedal/cluster/kmeans.py
Outdated
) | ||
|
||
use_raw_input = _get_config().get("use_raw_input") is True | ||
if use_raw_input and _get_sycl_namespace(X)[0] is not None: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
if use_raw_input and _get_sycl_namespace(X)[0] is not None: | |
if use_raw_input and sua_iface is not None: |
- move line 284 above this
/intelci: run |
1 similar comment
/intelci: run |
X, y, dtype=[np.float64, np.float32], accept_2d_y=True, force_all_finite=False | ||
) | ||
y = np.asarray(y, dtype=X.dtype) | ||
queue = self._queue = getattr(policy, "_queue", None) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
a little confused on this line - line 90 has policy set based on queue, then line 91 sets queue based on policy?
@@ -194,6 +195,9 @@ def _onedal_finalize_fit(self, queue=None): | |||
def _onedal_partial_fit(self, X, sample_weight=None, queue=None, check_input=True): | |||
first_pass = not hasattr(self, "n_samples_seen_") or self.n_samples_seen_ == 0 | |||
|
|||
use_raw_input = get_config()["use_raw_input"] | |||
# never check input when using raw input | |||
check_input &= use_raw_input is False |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ethan TODO: add this logic to other sklearnex incremental estimators
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
- also to _onedal_fit here
@@ -17,6 +17,7 @@ | |||
import logging | |||
|
|||
from daal4py.sklearn._utils import daal_check_version | |||
from sklearnex._config import get_config |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
minor comment - let's unify these eventually
from sklearnex._config import get_config | |
from .._config import get_config |
/intelci: run |
/intelci: run |
/intelci: run |
/intelci: run |
Description
Add a comprehensive description of proposed changes
List associated issue number(s) if exist(s): #6 (for example)
Documentation PR (if needed): #1340 (for example)
Benchmarks PR (if needed): IntelPython/scikit-learn_bench#155 (for example)
PR should start as a draft, then move to ready for review state after CI is passed and all applicable checkboxes are closed.
This approach ensures that reviewers don't spend extra time asking for regular requirements.
You can remove a checkbox as not applicable only if it doesn't relate to this PR in any way.
For example, PR with docs update doesn't require checkboxes for performance while PR with any change in actual code should have checkboxes and justify how this code change is expected to affect performance (or justification should be self-evident).
Checklist to comply with before moving PR from draft:
PR completeness and readability
Testing
Performance