Skip to content

Conversation

@shcheklein
Copy link
Contributor

@shcheklein shcheklein commented Apr 20, 2023

Adds an option to normalize confusion matrix like this:

live.log_sklearn_plot(
        "confusion_matrix",
        [0, 0, 1, 1],
        [0, 1, 1, 0],
        name="confusion_matrix_normalized",
        normalized=True,
    )

Quite handy to see a better heat map in case of classes being highly misbalanced.


Thank you for the contribution - we'll try to review it as soon as possible. 🙏

@shcheklein shcheklein force-pushed the add-normalized-cm-plot branch 3 times, most recently from 425c76d to b53c65f Compare April 20, 2023 23:36
@shcheklein shcheklein changed the title feat(sklplots): make sklearn plot properties instance level feat(sklplots): add confusion matrix normalized option Apr 20, 2023
@shcheklein shcheklein force-pushed the add-normalized-cm-plot branch from b53c65f to 5e8f77b Compare April 21, 2023 00:32
@shcheklein shcheklein requested a review from daavoo April 21, 2023 00:54
@shcheklein shcheklein marked this pull request as ready for review April 21, 2023 00:54
@daavoo daavoo added A: log_sklearn_plot Area: `live.log_sklearn_plot` feature labels Apr 21, 2023
@shcheklein
Copy link
Contributor Author

@daavoo @dberenbaum is it good to go? (I can create a doc PR then)

@dberenbaum dberenbaum merged commit bca4682 into main Apr 21, 2023
@dberenbaum dberenbaum deleted the add-normalized-cm-plot branch April 21, 2023 17:51
@shcheklein
Copy link
Contributor Author

Docs PR is here treeverse/dvc.org#4486

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

A: log_sklearn_plot Area: `live.log_sklearn_plot` feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants