Update wmvfts.py

This commit is contained in:
Petrônio Cândido de Lima e Silva 2023-05-26 14:44:46 -03:00 committed by GitHub
parent afc9b219ce
commit 4e0ee7cdd9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -69,7 +69,7 @@ class WeightedMVFTS(mvfts.MVFTS):
self.shortname = "WeightedMVFTS" self.shortname = "WeightedMVFTS"
self.name = "Weighted Multivariate FTS" self.name = "Weighted Multivariate FTS"
self.has_classification = True self.has_classification = True
self.class_weigths : dict = kwargs.get("class_weights", {}) self.class_weights : dict = kwargs.get("class_weights", {})
def generate_flrg(self, flrs): def generate_flrg(self, flrs):
@ -102,7 +102,7 @@ class WeightedMVFTS(mvfts.MVFTS):
for k,v in _flrg.RHS.items(): for k,v in _flrg.RHS.items():
classification[k] += (v / _flrg.count) * mb classification[k] += (v / _flrg.count) * mb
classification = activation(classification, self.class_weigths) classification = activation(classification, self.class_weights)
ret.append(classification) ret.append(classification)