diff --git a/pyFTS/models/multivariate/wmvfts.py b/pyFTS/models/multivariate/wmvfts.py index a944970..cc7a5b3 100644 --- a/pyFTS/models/multivariate/wmvfts.py +++ b/pyFTS/models/multivariate/wmvfts.py @@ -82,6 +82,8 @@ class WeightedMVFTS(mvfts.MVFTS): 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)