49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
from pyFTS.common import FuzzySet, Membership
|
|
import numpy as np
|
|
|
|
|
|
class Partitioner(object):
|
|
def __init__(self,name,data,npart,func = Membership.trimf, names=None, prefix="A", transformation=None):
|
|
self.name = name
|
|
self.partitions = npart
|
|
self.sets = []
|
|
self.membership_function = func
|
|
self.setnames = names
|
|
self.prefix = prefix
|
|
self.transformation = transformation
|
|
|
|
if transformation is not None:
|
|
ndata = transformation.apply(data)
|
|
else:
|
|
ndata = data
|
|
|
|
_min = min(ndata)
|
|
if _min < 0:
|
|
self.min = _min * 1.1
|
|
else:
|
|
self.min = _min * 0.9
|
|
|
|
_max = max(ndata)
|
|
if _max > 0:
|
|
self.max = _max * 1.1
|
|
else:
|
|
self.max = _max * 0.9
|
|
self.sets = self.build(ndata)
|
|
|
|
def build(self, data):
|
|
pass
|
|
|
|
def plot(self,ax):
|
|
ax.set_title(self.name)
|
|
ax.set_ylim([0, 1])
|
|
ax.set_xlim([self.min, self.max])
|
|
for s in self.sets:
|
|
if s.mf == Membership.trimf:
|
|
ax.plot([s.parameters[0], s.parameters[1], s.parameters[2]], [0, 1, 0])
|
|
elif s.mf == Membership.gaussmf:
|
|
tmpx = [kk for kk in np.arange(s.lower, s.upper)]
|
|
tmpy = [s.membership(kk) for kk in np.arange(s.lower, s.upper)]
|
|
ax.plot(tmpx, tmpy)
|
|
|
|
def __str__(self):
|
|
return self.name |