Skip to content

Fixed plot cutoff issue, added plot saving feature #107

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

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Fixed plot cutoff issue, added plot saving feature #107

wants to merge 1 commit into from

Conversation

harjotsinghparmar
Copy link

Was woking on a project and felt the need to have pdf/png outputs of the confusion matrix plot.
Have added tight layout and ability to save file using save parameter in the function.

@@ -256,6 +265,11 @@ def plot(self, normalized=False, backend='matplotlib',
elif backend == 'seaborn':
import seaborn as sns
ax = sns.heatmap(df, **kwargs)
# Saving plot
if save == True:
Copy link
Contributor

Choose a reason for hiding this comment

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

if save:

is enough

Copy link
Author

@harjotsinghparmar harjotsinghparmar left a comment

Choose a reason for hiding this comment

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

Makes sense.

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