瀏覽代碼

add display metric

Jérôme BUISINE 3 年之前
父節點
當前提交
9f17305ccc
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      find_best_attributes_surrogate_svm.py

+ 1 - 1
find_best_attributes_surrogate_svm.py

@@ -192,7 +192,7 @@ def main():
             diff = end - start
 
             print('----')
-            print("Real evaluation took: {}, score found: {}".format(divmod(diff.days * 86400 + diff.seconds, 60), test_roc_auc))
+            print("Real evaluation took: {}, score found: {}".format(divmod(diff.days * 86400 + diff.seconds, 60), 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))