Add files via upload

add knowledge about data type

predictive_modeling/class_predictive_configurator.py

@@ -21,7 +21,7 @@ def _dict_product(self, dicts):
         return (dict(zip(dicts, x)) for x in itertools.product(*dicts.values()))
 
 
-    def create_predictive_configs(self):
+    def create_predictive_configs(self, data_object):
 
         '''
         Creates list of predictive configurations
@@ -30,12 +30,16 @@ def create_predictive_configs(self):
         -------
             pred_configs (list of dictionaries) :  the predictive configurations
         '''
-
+
+        data_type = data_object.data_type
         pred_configs = []
         for pred_name, pred_info in self.pred_algs.items():
-            for v in self._dict_product(pred_info):
-                for fs_name, fs_info in self.fs_algs.items():
-                    for k in self._dict_product(fs_info):
-                        pred_configs.append({"pred_name": pred_name} | v | {"fs_name": fs_name} | k)
+            if data_type in pred_info['data_type']:
+                # remove data type info from output combinations
+                del pred_info['data_type']
+                for v in self._dict_product(pred_info):
+                    for fs_name, fs_info in self.fs_algs.items():
+                        for k in self._dict_product(fs_info):
+                            pred_configs.append({"pred_name": pred_name} | v | {"fs_name": fs_name} | k)
 
         return pred_configs