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