diff --git a/pyFTS/partitioners/Util.py b/pyFTS/partitioners/Util.py index 396eeca..ad40384 100644 --- a/pyFTS/partitioners/Util.py +++ b/pyFTS/partitioners/Util.py @@ -18,7 +18,7 @@ all_methods = [Grid.GridPartitioner, Entropy.EntropyPartitioner, FCM.FCMPartitio mfs = [Membership.trimf, Membership.gaussmf, Membership.trapmf] -def plot_sets(data, sets: dict, titles : list, size=[12, 10], save=False, file=None, axis=None): +def plot_sets(sets: dict, titles : list, size=[12, 10], save=False, file=None, axis=None): """ Plot all fuzzy sets in a Partitioner @@ -26,7 +26,7 @@ def plot_sets(data, sets: dict, titles : list, size=[12, 10], save=False, file=N num = len(sets) if axis is None: - fig, axes = plt.subplots(nrows=num, ncols=1,figsize=size) + fig, axes = plt.subplots(nrows=num, ncols=1, figsize=size) for k in np.arange(0,num): ticks = [] x = [] @@ -54,7 +54,7 @@ def plot_sets(data, sets: dict, titles : list, size=[12, 10], save=False, file=N Util.show_and_save_image(fig, file, save) -def plot_partitioners(data, objs, tam=[12, 10], save=False, file=None, axis=None): +def plot_partitioners(objs, tam=[12, 10], save=False, file=None, axis=None): sets = [k.sets for k in objs] titles = [k.name for k in objs] plot_sets(sets, titles, tam, save, file, axis)