X and y fixed in regression lexicase

alcides · Nov 1, 2024 · aa263bd · aa263bd
1 parent 285d55d
commit aa263bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/benchmarks/regression_lexicase.py b/examples/benchmarks/regression_lexicase.py
@@ -23,6 +23,8 @@ def __init__(self, X, y, feature_names):
     def setup_problem(self, data, target):
 
         # Problem
+        X = data
+        y = target
         n_cases = 50
         case_size = int(len(data) / n_cases)
 
@@ -44,8 +46,6 @@ def calculate_grouped_errors(pred_error, n_cases, case_size):
             return grouped_errors
 
         def lexicase_fitness_function(n: Expression):
-            X = data
-            y = target
 
             try:
                 y_pred = forward_dataset(n.to_numpy(), X)