Jérôme BUISINE 3 лет назад
Родитель
Сommit
0224e7f541
1 измененных файлов с 4 добавлено и 4 удалено
  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)