Update wmvfts.py
This commit is contained in:
parent
afc9b219ce
commit
73769ab5a7
@ -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)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user