Improvements of FCM_FTS

2020-01-28 15:59:03 -03:00 · 2020-01-28 15:59:03 -03:00 · 18a95dad2e
commit 18a95dad2e
parent be119604a9
1 changed files with 3 additions and 0 deletions
--- a/pyFTS/fcm/fts.py
+++ b/pyFTS/fcm/fts.py
@ -11,6 +11,9 @@ class FCM_FTS(hofts.HighOrderFTS):
        self.fcm = common.FuzzyCognitiveMap(**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)
        self.fcm.weights = ret['weights']