Update wmvfts.py - class_weights

PYFTS · May 26, 2023 · b8cfea5 · b8cfea5
1 parent 02f6022
commit b8cfea5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pyFTS/models/multivariate/wmvfts.py b/pyFTS/models/multivariate/wmvfts.py
@@ -69,6 +69,8 @@ def __init__(self, **kwargs):
         self.shortname = "WeightedMVFTS"
         self.name = "Weighted Multivariate FTS"
         self.has_classification = True
+        self.class_weigths : dict = kwargs.get("class_weights", {})
+
 
     def generate_flrg(self, flrs):
         for flr in flrs:
@@ -98,7 +100,7 @@ def classify(self, data, **kwargs):
                     for k,v in _flrg.RHS.items():
                         classification[k] += (v / _flrg.count) * mb
 
-            classification = activation(classification)
+            classification = activation(classification, self.class_weigths)
 
             ret.append(classification)