Source code for pyFTS.models.multivariate.wmvfts

from pyFTS.common import fts, FuzzySet, FLR, Membership, tree
from pyFTS.partitioners import Grid
from pyFTS.models.multivariate import mvfts, FLR as MVFLR, common, flrg as mvflrg

import numpy as np
import pandas as pd


class WeightedFLRG(mvflrg.FLRG):
    """
    Weighted Multivariate Fuzzy Logical Rule Group
    """

    def __init__(self, **kwargs):
        super(WeightedFLRG, self).__init__(**kwargs)
        self.order = kwargs.get('order', 1)
        self.LHS = kwargs.get('lhs', {})
        self.RHS = {}
        self.count = 0.0
        self.w = None

    def append_rhs(self, fset, **kwargs):
        if fset not in self.RHS:
            self.RHS[fset] = 1.0
        else:
            self.RHS[fset] += 1.0
        self.count += 1.0

    def weights(self):
        if self.w is None:
            self.w = np.array([self.RHS[c] / self.count for c in self.RHS.keys()])
        return self.w

    def get_midpoint(self, sets):
        mp = np.array([sets[c].centroid for c in self.RHS.keys()])
        return mp.dot(self.weights())


    def __str__(self):
        _str = ""
        for k in self.RHS.keys():
            _str += ", " if len(_str) > 0 else ""
            _str += k + " (" + str(round( self.RHS[k] / self.count, 3)) + ")"

        return self.get_key() + " -> " + _str


class WeightedMVFTS(mvfts.MVFTS):
    """
    Weighted Multivariate FTS
    """
    def __init__(self, **kwargs):
        super(WeightedMVFTS, self).__init__(order=1, **kwargs)
        self.explanatory_variables = []
        self.target_variable = None
        self.flrgs = {}
        self.is_multivariate = True
        self.shortname = "WeightedMVFTS"
        self.name = "Weighted Multivariate FTS"

    def generate_flrg(self, flrs):
        for flr in flrs:
            flrg = WeightedFLRG(lhs=flr.LHS)

            if flrg.get_key() not in self.flrgs:
                self.flrgs[flrg.get_key()] = flrg

            self.flrgs[flrg.get_key()].append_rhs(flr.RHS)