Source code for pyFTS.models.multivariate.common
import numpy as np
import pandas as pd
from pyFTS.common import FuzzySet, Composite
[docs]class MultivariateFuzzySet(Composite.FuzzySet):
"""
Multivariate Composite Fuzzy Set
"""
def __init__(self, name, **kwargs):
"""
Create an empty composite fuzzy set
:param name: fuzzy set name
"""
super(MultivariateFuzzySet, self).__init__(name)
self.sets = {}
self.target_variable = kwargs.get('target_variable',None)
[docs] def append_set(self, variable, set):
"""
Appends a new fuzzy set from a new variable
:param variable: an multivariate.variable instance
:param set: an common.FuzzySet instance
"""
self.sets[variable] = set
if variable == self.target_variable.name:
self.centroid = set.centroid
[docs] def membership(self, x):
mv = []
for var in self.sets.keys():
data = x[var]
mv.append(self.sets[var].membership(data))
return np.nanmin(mv)
[docs]def fuzzyfy_instance(data_point, var):
fsets = FuzzySet.fuzzyfy(data_point, var.partitioner, mode='sets', method='fuzzy', alpha_cut=var.alpha_cut)
return [(var.name, fs) for fs in fsets]
[docs]def fuzzyfy_instance_clustered(data_point, cluster, alpha_cut=0.0):
fsets = []
for fset in cluster.knn(data_point):
if cluster.sets[fset].membership(data_point) > alpha_cut:
fsets.append(fset)
return fsets