50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import numpy as np
|
|
from pyFTS.common import FuzzySet
|
|
|
|
|
|
class FLR(object):
|
|
def __init__(self, LHS, RHS):
|
|
self.LHS = LHS
|
|
self.RHS = RHS
|
|
|
|
def __str__(self):
|
|
return self.LHS.name + " -> " + self.RHS.name
|
|
|
|
|
|
class IndexedFLR(FLR):
|
|
def __init__(self, index, LHS, RHS):
|
|
super(IndexedFLR, self).__init__(LHS, RHS)
|
|
self.index = index
|
|
|
|
def __str__(self):
|
|
return str(self.index) + ": "+ self.LHS.name + " -> " + self.RHS.name
|
|
|
|
|
|
def generateNonRecurrentFLRs(fuzzyData):
|
|
flrs = {}
|
|
for i in range(2,len(fuzzyData)):
|
|
tmp = FLR(fuzzyData[i-1],fuzzyData[i])
|
|
flrs[str(tmp)] = tmp
|
|
ret = [value for key, value in flrs.items()]
|
|
return ret
|
|
|
|
|
|
def generateRecurrentFLRs(fuzzyData):
|
|
flrs = []
|
|
for i in np.arange(1,len(fuzzyData)):
|
|
flrs.append(FLR(fuzzyData[i-1],fuzzyData[i]))
|
|
return flrs
|
|
|
|
|
|
def generateIndexedFLRs(sets, indexer, data):
|
|
flrs = []
|
|
index = indexer.get_season_of_data(data)
|
|
ndata = indexer.get_data(data)
|
|
for k in np.arange(1,len(data)):
|
|
lhs = FuzzySet.getMaxMembershipFuzzySet(ndata[k-1],sets)
|
|
rhs = FuzzySet.getMaxMembershipFuzzySet(ndata[k], sets)
|
|
season = index[k]
|
|
flr = IndexedFLR(season,lhs,rhs)
|
|
flrs.append(flr)
|
|
return flrs
|