eval loss of math. test functions instead of score

SimonBlanke · Jan 2, 2024 · b926b10 · b926b10
1 parent 3826e15
commit b926b10
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/collect_search_data.py b/collect_search_data.py
@@ -1,16 +1,16 @@
 from surfaces import mathematical_functions, machine_learning_functions
 
+
 for mathematical_function in mathematical_functions:
     print("\n ---", mathematical_function.name, "---")
     try:
-        test_function = mathematical_function()
+        test_function = mathematical_function(metric="loss")
     except TypeError:
-        test_function = mathematical_function(n_dim=2)
+        test_function = mathematical_function(n_dim=2, metric="loss")
 
     test_function.collect_data()
 
-"""
+
 for machine_learning_function in machine_learning_functions:
     test_function = machine_learning_function()
     test_function.collect_data()
-"""