From 18a95dad2e823e8347c21d7142ed4c2e99ebdc59 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=C3=B4nio=20C=C3=A2ndido?= <petronio.candido@gmail.com>
Date: Tue, 28 Jan 2020 15:59:03 -0300
Subject: [PATCH] Improvements of FCM_FTS

---
 pyFTS/fcm/fts.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyFTS/fcm/fts.py b/pyFTS/fcm/fts.py
index cbeeff7..5fe05ce 100644
--- 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']