Parcourir la source

Merge branch 'release/v0.2.9'

Jérôme BUISINE il y a 4 ans
Parent
commit
0224e7f541
1 fichiers modifiés avec 4 ajouts et 4 suppressions
  1. 4 4
      find_best_attributes.py

+ 4 - 4
find_best_attributes.py

@@ -50,10 +50,6 @@ def validator(solution):
 
 
     return True
     return True
 
 
-# init solution (26 attributes)
-def init():
-    return BinarySolution([], 26).random(validator)
-
 def loadDataset(filename):
 def loadDataset(filename):
 
 
     ########################
     ########################
@@ -150,6 +146,10 @@ def main():
 
 
         return test_roc_auc
         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):
     if not os.path.exists(cfg.output_backup_folder):
         os.makedirs(cfg.output_backup_folder)
         os.makedirs(cfg.output_backup_folder)