-
-
Notifications
You must be signed in to change notification settings - Fork 26k
FEA Add array API support for GaussianMixture #30777
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
Conversation
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
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.
Overall looks good. I left a few comments
|
||
|
||
# TODO What is the expected behavior when weights init | ||
# and X are not in the same namespace/device? |
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.
I think this is not resolved yet. Can we remove the commented out code?
… are passed in + fixes
@lesteve Just one test failing and that has to do with array api strict on device and float32. Maybe we need to increase the tolerance further for this specific scenario. |
My honest impression is that these tests are fragile on float32 data but I don't really know if there is much we can do to improve the situation ... Even for On a GPU VM I also saw some test failures (a few more than in the CI actually) and raised the |
I don't think we can do much with trying to improve array-api-strict tests for float32 especially with respect to accuracy. As long as array-api-strict works generally I think that should be sufficient. |
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.
LGTM. Thank you for the work done in this PR @lesteve and @StefanieSenger
Thanks for the reviews @OmarManzoor and @ogrisel! One of the remaining question in the old and long TODO list: should we implement
scikit-learn/sklearn/decomposition/_pca.py Lines 848 to 857 in 8792943
I am guessing the array_api tags is only used for the common tests right now, right? |
Good questions:
|
Working on it with @StefanieSenger.
Link to TODO