36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
import numpy as np
|
|
import math
|
|
import random as rnd
|
|
import functools, operator
|
|
from pyFTS.common import FuzzySet, Membership
|
|
from pyFTS.partitioners import partitioner
|
|
|
|
|
|
class GridPartitioner(partitioner.Partitioner):
|
|
"""Even Length Grid Partitioner"""
|
|
def __init__(self, data, npart, func = Membership.trimf, transformation=None):
|
|
super(GridPartitioner, self).__init__("Grid", data, npart, func=func, transformation=transformation)
|
|
|
|
def build(self, data):
|
|
sets = []
|
|
|
|
dlen = self.max - self.min
|
|
partlen = dlen / self.partitions
|
|
|
|
count = 0
|
|
for c in np.arange(self.min, self.max, partlen):
|
|
if self.membership_function == Membership.trimf:
|
|
sets.append(
|
|
FuzzySet.FuzzySet(self.prefix + str(count), Membership.trimf, [c - partlen, c, c + partlen],c))
|
|
elif self.membership_function == Membership.gaussmf:
|
|
sets.append(
|
|
FuzzySet.FuzzySet(self.prefix + str(count), Membership.gaussmf, [c, partlen / 3], c))
|
|
elif self.membership_function == Membership.trapmf:
|
|
q = partlen / 2
|
|
sets.append(
|
|
FuzzySet.FuzzySet(self.prefix + str(count), Membership.trapmf, [c - partlen, c - q, c + q, c + partlen], c))
|
|
count += 1
|
|
|
|
|
|
return sets
|