diff --git a/pyFTS/nonstationary/common.py b/pyFTS/nonstationary/common.py index 3f6fd0c..07bb7b0 100644 --- a/pyFTS/nonstationary/common.py +++ b/pyFTS/nonstationary/common.py @@ -7,20 +7,21 @@ IEEE Transactions on Fuzzy Systems, v. 16, n. 4, p. 1072-1086, 2008. import numpy as np from pyFTS import * -from pyFTS.common import FuzzySet, Membership +from pyFTS.common import FuzzySet as FS, Membership from pyFTS.partitioners import partitioner from pyFTS.nonstationary import perturbation -class MembershipFunction(object): +class FuzzySet(FS.FuzzySet): """ - Non Stationary Membership Function + Non Stationary Fuzzy Sets """ + def __init__(self, name, mf, parameters, **kwargs): """ - Non Stationary Membership Function + Constructor :param name: - :param mf: + :param mf: Fuzzy Membership Function :param parameters: :param kwargs: - location: Pertubation function that affects the location of the membership function @@ -30,8 +31,8 @@ class MembershipFunction(object): - noise: Pertubation function that adds noise on the membership function - noise_params: Parameters for noise pertubation function """ - self.mf = mf - self.parameters = parameters + super(FuzzySet, self).__init__(name=name, mf=mf, parameters=parameters, centroid=None) + self.location = kwargs.get("location", None) self.location_params = kwargs.get("location_params", None) self.location_roots = kwargs.get("location_roots", 0) @@ -41,97 +42,85 @@ class MembershipFunction(object): self.noise = kwargs.get("noise", None) self.noise_params = kwargs.get("noise_params", None) self.perturbated_parameters = {} - - if self.location is not None and not isinstance(self.location, (list,set)): + + if self.location is not None and not isinstance(self.location, (list, set)): self.location = [self.location] self.location_params = [self.location_params] self.location_roots = [self.location_roots] - + if self.width is not None and not isinstance(self.width, (list, set)): self.width = [self.width] self.width_params = [self.width_params] self.width_roots = [self.width_roots] - + def perform_location(self, t, param): if self.location is None: return param - + l = len(self.location) - - inc = sum([self.location[k](t + self.location_roots[k], self.location_params[k]) for k in np.arange(0,l)] ) - + + inc = sum([self.location[k](t + self.location_roots[k], self.location_params[k]) for k in np.arange(0, l)]) + if self.mf == Membership.gaussmf: - #changes only the mean parameter + # changes only the mean parameter return [param[0] + inc, param[1]] elif self.mf == Membership.sigmf: - #changes only the midpoint parameter + # changes only the midpoint parameter return [param[0], param[1] + inc] elif self.mf == Membership.bellmf: return [param[0], param[1], param[2] + inc] else: - #translate all parameters + # translate all parameters return [k + inc for k in param] - + def perform_width(self, t, param): if self.width is None: return param - + l = len(self.width) - + inc = sum([self.width[k](t + self.width_roots[k], self.width_params[k]) for k in np.arange(0, l)]) - + if self.mf == Membership.gaussmf: - #changes only the variance parameter + # changes only the variance parameter return [param[0], param[1] + inc] elif self.mf == Membership.sigmf: - #changes only the smooth parameter + # changes only the smooth parameter return [param[0] + inc, param[1]] elif self.mf == Membership.trimf: - tmp = inc/2 + tmp = inc / 2 return [param[0] - tmp, param[1], param[2] + tmp] elif self.mf == Membership.trapmf: - l = (param[3]-param[0]) + l = (param[3] - param[0]) rab = (param[1] - param[0]) / l rcd = (param[3] - param[2]) / l - return [param[0] - inc, param[1] - inc*rab, param[2] + inc*rcd, param[3] + inc] + return [param[0] - inc, param[1] - inc * rab, param[2] + inc * rcd, param[3] + inc] else: return param - + def membership(self, x, t): """ Calculate the membership value of a given input :param x: input value :return: membership value of x at this fuzzy set """ - + self.perturbate_parameters(t) - + tmp = self.mf(x, self.perturbated_parameters[t]) - + if self.noise is not None: tmp += self.noise(t, self.noise_params) - + return tmp - + def perturbate_parameters(self, t): if t not in self.perturbated_parameters: param = self.parameters param = self.perform_location(t, param) param = self.perform_width(t, param) self.perturbated_parameters[t] = param - - def __str__(self): - tmp = "" - if self.location is not None: - tmp += "Loc. Pert.: " - for ct, f in enumerate(self.location): - tmp += str(f.__name__) + "(" + str(self.location_params[ct]) + ") " - if self.width is not None: - tmp += "Wid. Pert.: " - for ct, f in enumerate(self.width): - tmp += str(f.__name__) + "(" + str(self.width_params[ct]) + ") " - return str(self.mf.__name__) + "(" + str(self.parameters) + ") " + tmp - + def get_midpoint(self, t): self.perturbate_parameters(t) @@ -180,26 +169,17 @@ class MembershipFunction(object): else: return param - - -class FuzzySet(FuzzySet.FuzzySet): - """ - Non Stationary Fuzzy Sets - """ - - def __init__(self, name, mf, **kwargs): - """ - Constructor - :param name: Fuzzy Set name - :param mf: NonStationary Membership Function - """ - super(FuzzySet, self).__init__(name=name, mf=mf, parameters=None, centroid=None) - - def membership(self, x, t): - return self.mf.membership(x,t) - def __str__(self): - return self.name + ": " + str(self.mf) + tmp = "" + if self.location is not None: + tmp += "Loc. Pert.: " + for ct, f in enumerate(self.location): + tmp += str(f.__name__) + "(" + str(self.location_params[ct]) + ") " + if self.width is not None: + tmp += "Wid. Pert.: " + for ct, f in enumerate(self.width): + tmp += str(f.__name__) + "(" + str(self.width_params[ct]) + ") " + return self.name + ": " + str(self.mf.__name__) + "(" + str(self.parameters) + ") " + tmp class PolynomialNonStationaryPartitioner(partitioner.Partitioner): @@ -221,14 +201,13 @@ class PolynomialNonStationaryPartitioner(partitioner.Partitioner): for ct, set in enumerate(part.sets): loc_roots = np.roots(loc_params[ct])[0] wid_roots = np.roots(wid_params[ct])[0] - mf = MembershipFunction(name=set.name, mf=set.mf, parameters=set.parameters, - location=perturbation.polynomial, - location_params=loc_params[ct], - location_roots=loc_roots, #**kwargs) - width=perturbation.polynomial, - width_params=wid_params[ct], - width_roots=wid_roots, **kwargs) - tmp = FuzzySet(set.name, mf, **kwargs) + tmp = FuzzySet(set.name, set.mf, set.parameters, + location=perturbation.polynomial, + location_params=loc_params[ct], + location_roots=loc_roots, #**kwargs) + width=perturbation.polynomial, + width_params=wid_params[ct], + width_roots=wid_roots, **kwargs) self.sets.append(tmp) diff --git a/pyFTS/nonstationary/util.py b/pyFTS/nonstationary/util.py index 5243954..3da746c 100644 --- a/pyFTS/nonstationary/util.py +++ b/pyFTS/nonstationary/util.py @@ -12,9 +12,9 @@ def plot_sets(uod, sets, start=0, end=10, tam=[5, 5], colors=None, save=False, f for t in np.arange(start,end,1): for ct, set in enumerate(sets): set.membership(0, t) - param = set.mf.perturbated_parameters[t] + param = set.perturbated_parameters[t] - if set.mf.mf == Membership.trimf: + if set.mf == Membership.trimf: axes.plot([t, t+1, t], param) ticks.extend(["t+"+str(t),""])