Skip to content

[bugfix] Normalize Decision Tree .values attribute to match sklearn #2377

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

Merged
merged 5 commits into from
Mar 26, 2025

Conversation

icfaust
Copy link
Contributor

@icfaust icfaust commented Mar 24, 2025

Description

The values attribute (https://scikit-learn.org/stable/auto_examples/tree/plot_unveil_tree_structure.html#what-is-the-values-array-used-here) is normalized (i.e. the fraction of samples), where we had previously set it to the absolute number of samples. A division after a sum is applied at the end in order to conform to sklearn and to try and fix the error observed in #1919.

A second error is mentioned in the issue where the classes aren't properly stored in the DecisionTree, a fix is made to use all of self.classes_ instead of self.classes_[0].


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

  • I have reviewed my changes thoroughly before submitting this pull request.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have updated the documentation to reflect the changes or created a separate PR with update and provided its number in the description, if necessary.
  • Git commit message contains an appropriate signed-off-by string (see CONTRIBUTING.md for details).
  • I have added a respective label(s) to PR if I have a permission for that.
  • I have resolved any merge conflicts that might occur with the base branch.

Testing

  • I have run it locally and tested the changes extensively.
  • All CI jobs are green or I have provided justification why they aren't.
  • I have extended testing suite if new functionality was introduced in this PR.

Performance

  • I have measured performance for affected algorithms using scikit-learn_bench and provided at least summary table with measured data, if performance change is expected.
  • I have provided justification why performance has changed or why changes are not expected.
  • I have provided justification why quality metrics have changed or why changes are not expected.
  • I have extended benchmarking suite and provided corresponding scikit-learn_bench PR if new measurable functionality was introduced in this PR.

Copy link

codecov bot commented Mar 24, 2025

Codecov Report

Attention: Patch coverage is 42.85714% with 4 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
onedal/primitives/tree_visitor.cpp 33.33% 0 Missing and 4 partials ⚠️
Flag Coverage Δ
azure 78.04% <100.00%> (-0.01%) ⬇️
github 70.54% <42.85%> (+0.10%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Files with missing lines Coverage Δ
sklearnex/ensemble/_forest.py 70.32% <100.00%> (-0.05%) ⬇️
onedal/primitives/tree_visitor.cpp 80.39% <33.33%> (-1.37%) ⬇️

... and 3 files with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@icfaust
Copy link
Contributor Author

icfaust commented Mar 24, 2025

/intelci: run

@icfaust
Copy link
Contributor Author

icfaust commented Mar 25, 2025

/intelci: run

@icfaust icfaust marked this pull request as ready for review March 25, 2025 13:21
@icfaust icfaust merged commit 87f8cf6 into uxlfoundation:main Mar 26, 2025
31 of 32 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants