-
Notifications
You must be signed in to change notification settings - Fork 2
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
Use scikit-learn confusion matrix instead of own custom code #20
Conversation
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.
Hi Johannes @jo-mueller ,
just some minor stuff and I'm suggesting to remove one function. Furthermore, could you please also delete overlap_matrix_benchmarking.ipynb
? I created this branch with that notebook just to demonstrate something. I wasn't thinking of making this notebook part of the repository.
I'm happy to merge your changes afterwards.
Thanks!
Best,
Robert
Co-authored-by: Robert Haase <haesleinhuepf@users.noreply.github.com>
Co-authored-by: Robert Haase <haesleinhuepf@users.noreply.github.com>
The functionality is now directly implemented in the _intersection_over_union.py
Hi Johannes @jo-mueller , this all looks good to me. Shall I merge? |
Yes :) |
Wonderful. Thanks for working on this @jo-mueller ! |
Enhanced overlap matrix calculation with sklearn implementation
The calculation of the label overlap matrix has been exchanged with the implementation from sklearn, which is considerably faster. After the upgrade, the benchmark (/notebooks/label/overlap_matrix_benchmarking.ipynb)) shows comparable results for both approaches.
Type of change
References
Closes #3
Tests
Final checks