sam/pyFTS
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

Update wmvfts.py - class_weights

Browse Source
This commit is contained in:
Petrônio Cândido de Lima e Silva 2023-05-26 14:24:41 -03:00 committed by GitHub
parent 560dcfacf6
commit afc9b219ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyFTS/models/multivariate/wmvfts.py
View File

@ -82,6 +82,8 @@ class WeightedMVFTS(mvfts.MVFTS):
self.flrgs[flrg.get_key()].append_rhs(flr.RHS) self.flrgs[flrg.get_key()].append_rhs(flr.RHS)
def classify(self, data, **kwargs): 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 = [] ret = []
ndata = self.apply_transformations(data) ndata = self.apply_transformations(data)
activation = kwargs.get('activation', Activations.scale) activation = kwargs.get('activation', Activations.scale)