Update wmvfts.py - class_weights

PYFTS · May 26, 2023 · afc9b21 · afc9b21
1 parent 560dcfa
commit afc9b21
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyFTS/models/multivariate/wmvfts.py b/pyFTS/models/multivariate/wmvfts.py
@@ -82,6 +82,8 @@ def generate_flrg(self, flrs):
             self.flrgs[flrg.get_key()].append_rhs(flr.RHS)
 
     def classify(self, data, **kwargs):
+        if len(self.class_weights) == 0:
+            self.class_weights = {k : 1.0 for k in self.target_variable.partitioner.sets.keys()}
         ret = []
         ndata = self.apply_transformations(data)
         activation = kwargs.get('activation', Activations.scale)