Przeglądaj źródła

dynamic size of solution when finding best attributes

Jérôme BUISINE 4 lat temu
rodzic
commit
04742934ca
1 zmienionych plików z 4 dodań i 4 usunięć
  1. 4 4
      find_best_attributes.py

+ 4 - 4
find_best_attributes.py

@@ -50,10 +50,6 @@ def validator(solution):
 
     return True
 
-# init solution (26 attributes)
-def init():
-    return BinarySolution([], 26).random(validator)
-
 def loadDataset(filename):
 
     ########################
@@ -150,6 +146,10 @@ def main():
 
         return test_roc_auc
 
+    # init solution (`n` attributes)
+    def init():
+        return BinarySolution([], number_of_values).random(validator)
+
     if not os.path.exists(cfg.output_backup_folder):
         os.makedirs(cfg.output_backup_folder)