steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

requirements

Browse Source
This commit is contained in:
Si11ium 2020-05-14 23:08:35 +02:00
parent 68084d42a2
commit 2c4d903fa2
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
visualization/tools.py
View File

@ -6,11 +6,14 @@ class Plotter(object):
def __init__(self, root_path=''):
self.root_path = Path(root_path)
def save_current_figure(self, path, extention='.png'):
def save_current_figure(self, path, extention='.png', naked=True):
fig, _ = plt.gcf(), plt.gca()
# Prepare save location and check img file extention
path = self.root_path / Path(path if str(path).endswith(extention) else f'{str(path)}{extention}')
path.parent.mkdir(exist_ok=True, parents=True)
if naked:
plt.axis('off')
fig.savefig(path, bbox_inches='tight', transparent=True, pad_inches=0)
fig.savefig(path)
fig.clf()