Improvements of FCM_FTS

This commit is contained in:
Petrônio Cândido 2020-01-28 15:59:03 -03:00
parent be119604a9
commit 18a95dad2e

View File

@ -11,6 +11,9 @@ class FCM_FTS(hofts.HighOrderFTS):
self.fcm = common.FuzzyCognitiveMap(**kwargs) self.fcm = common.FuzzyCognitiveMap(**kwargs)
def train(self, data, **kwargs): def train(self, data, **kwargs):
GA.parameters['num_concepts'] = self.partitioner.partitions
GA.parameters['order'] = self.order
GA.parameters['partitioner'] = self.partitioner
ret = GA.execute(data, **kwargs) ret = GA.execute(data, **kwargs)
self.fcm.weights = ret['weights'] self.fcm.weights = ret['weights']