Fixed target

geml/common.py

@@ -120,16 +120,16 @@ def fit(self, X, y):
         random = NativeRandomSource(self.seed)
 
         feature_names, data = self.prepare_inputs(X)
-        target = self.prepare_outputs(y)
-        assert data.shape[0] == target.shape[0]
+        y = self.prepare_outputs(y)
+        assert data.shape[0] == y.shape[0]
 
-        grammar = self.get_grammar(feature_names, data, target)
+        grammar = self.get_grammar(feature_names, data, y)
 
         def fitness_function(x: Expression) -> float:
             try:
                 y_pred = forward_dataset(x.to_numpy(), data)
                 with np.errstate(all="ignore"):
-                    return r2_score(target, y_pred)
+                    return r2_score(y, y_pred)
             except ValueError:
                 return -10000000
 

geneticengine/algorithms/hill_climbing.py

@@ -28,7 +28,6 @@ def __init__(
     def search(self) -> Individual:
         assert isinstance(self.representation, RepresentationWithMutation)
         current_ind = None
-        print(self.is_done())
         while not self.is_done():
             if current_ind is None:
                 n = self.representation.create_genotype(self.random)
@@ -41,5 +40,4 @@ def search(self) -> Individual:
                 neighbourhood = [Individual(genotype=n2, representation=self.representation) for n2 in genotypes]
                 self.tracker.evaluate(neighbourhood)
             current_ind = self.tracker.get_best_individual()
-        print("debug", current_ind, self.tracker.get_best_individual())
         return self.tracker.get_best_individual()