41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import numpy as np
|
|
from pyFTS import *
|
|
from pyFTS.common import Membership
|
|
|
|
|
|
class FuzzySet:
|
|
def __init__(self, name, mf, parameters, centroid):
|
|
self.name = name
|
|
self.mf = mf
|
|
self.parameters = parameters
|
|
self.centroid = centroid
|
|
if self.mf == Membership.trimf:
|
|
self.lower = min(parameters)
|
|
self.upper = max(parameters)
|
|
elif self.mf == Membership.gaussmf:
|
|
self.lower = parameters[0] - parameters[1]*3
|
|
self.upper = parameters[0] + parameters[1]*3
|
|
|
|
def membership(self, x):
|
|
return self.mf(x, self.parameters)
|
|
|
|
def __str__(self):
|
|
return self.name + ": " + str(self.mf) + "(" + str(self.parameters) + ")"
|
|
|
|
|
|
def fuzzyInstance(inst, fuzzySets):
|
|
mv = np.array([fs.membership(inst) for fs in fuzzySets])
|
|
return mv
|
|
|
|
|
|
def getMaxMembershipFuzzySet(inst, fuzzySets):
|
|
mv = fuzzyInstance(inst, fuzzySets)
|
|
return fuzzySets[np.argwhere(mv == max(mv))[0, 0]]
|
|
|
|
|
|
def fuzzySeries(data, fuzzySets):
|
|
fts = []
|
|
for item in data:
|
|
fts.append(getMaxMembershipFuzzySet(item, fuzzySets))
|
|
return fts
|