Ver código fonte

Fix pow criterion

Jérôme BUISINE 3 anos atrás
pai
commit
a6287e17df
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      find_best_attributes_surrogate_svm.py

+ 1 - 1
find_best_attributes_surrogate_svm.py

@@ -194,7 +194,7 @@ def main():
             print('----')
             print("Real evaluation took: {}, score found: {}".format(divmod(diff.days * 86400 + diff.seconds, 60), test_roc_auc))
 
-            return test_roc_auc * (1 - math.pow(test_roc_auc - train_roc_auc), 2)
+            return test_roc_auc * (1 - math.pow(test_roc_auc - train_roc_auc, 2))
 
 
     # build all output folder and files based on `output` name