sam/pyFTS
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

Improvements in nonstationary.Utils.plot_sets

Browse Source
This commit is contained in:
Petrônio Cândido de Lima e Silva 2020-10-05 10:28:53 -03:00 committed by GitHub
parent 57f9e2f6f1
commit 789734de51
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyFTS/models/nonstationary/util.py
View File

@ -7,7 +7,7 @@ from pyFTS.common import Membership, Util
def plot_sets(partitioner, start=0, end=10, step=1, tam=[5, 5], colors=None,
save=False, file=None, axes=None, data=None, window_size = 1, only_lines=False):
save=False, file=None, axes=None, data=None, window_size = 1, only_lines=False, legend=True):
range = np.arange(start,end,step)
ticks = []
@ -43,14 +43,15 @@ def plot_sets(partitioner, start=0, end=10, step=1, tam=[5, 5], colors=None,
axes.set_xlabel("Time")
plt.xticks([k for k in range], ticks, rotation='vertical')
if legend:
handles0, labels0 = axes.get_legend_handles_labels()
lgd = axes.legend(handles0, labels0, loc=2, bbox_to_anchor=(1, 1))
if data is not None:
axes.plot(np.arange(start, start + len(data), 1), data,c="black")
if file is not None:
plt.tight_layout()
Util.show_and_save_image(fig, file, save)