Skip to content
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

switch cupy_conf_mat to confusion_matrix #191

Merged
merged 1 commit into from
Mar 1, 2021

Conversation

beckernick
Copy link
Member

This PR:

  • Rips out the custom confusion matrix utility and uses cuml.metrics.confusion_matrix
  • Verified correct

See rapidsai/cuml#3567 for the need to explicitly cast to int32. After that issue is resolved, we can update this.

@beckernick beckernick self-assigned this Mar 1, 2021
@beckernick beckernick changed the title switch cup_conf_mat to confusion_matrix switch cupy_conf_mat to confusion_matrix Mar 1, 2021
Copy link
Member

@ayushdg ayushdg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the update @beckernick , In a followup PR the confusion matrix logic for q28 can be updated as well. ( I believe that's a distributed dask cm) which should be supported based on #rapidsai/cuml#1817

@beckernick
Copy link
Member Author

Ah, excellent. Let's do that, too.

@beckernick beckernick merged commit f97a7cf into rapidsai:main Mar 1, 2021
@beckernick beckernick deleted the confusion-matrix-cleanup branch April 26, 2021 15:45
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