@@ -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