-
Notifications
You must be signed in to change notification settings - Fork 606
ENH: Add adjusted_mutual_info_score metric #7418
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
ENH: Add adjusted_mutual_info_score metric #7418
Conversation
47e295e to
325fb22
Compare
77d5494 to
88da3ce
Compare
|
Rebased this branch with the latest main and force-pushed to update the commit history. The implementation for adjusted_mutual_info_score is stable and aligned with scikit-learn’s behavior. @bdice @csadorf @divyegala — please review when convenient. Thanks for your guidance and feedback! |
|
Have a look at the diff and commits on this PR, something must have gone wrong with the rebase. |
88da3ce to
1e7eb53
Compare
|
This PR has significant file organization issues that need cleanup. The same code appears multiple times across different files - including files with invalid extensions like The PR also includes both a CPU-based sklearn wrapper and a separate GPU implementation without clarity on which is intended or how they integrate. Given the extent of cleanup needed, I'm going to close this PR. Please open a new PR with only the necessary files in their correct locations and a clear implementation approach. |
Summary
This PR implements the
adjusted_mutual_info_scoremetric in cuML’s cluster metrics module.It mirrors the scikit-learn API for consistent behavior and prepares for future GPU acceleration.
Related Issue
Fixes: #7294
Implementation Details
python/cuml/metrics/cluster/adjusted_mutual_info_score.py__init__.pyto expose the new metricChecklist
Author: Vardhan Naidu (@vardhan30016)