Fixed wrong context being used to update crossover

cavalab · Oct 3, 2024 · fb776c4 · fb776c4
1 parent 4ccf09c
commit fb776c4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/vary/variation.h b/src/vary/variation.h
@@ -231,20 +231,23 @@ class Variation {
                 *r.select_randomly(parents.begin(), parents.end())];
 
             vector<Individual<T>> ind_parents;
-            VectorXf context = get_context(mom.program.Tree,
+            VectorXf root_context = get_context(mom.program.Tree,
                                            mom.program.Tree.begin());
-
-            string choice = this->variation_bandit.choose(context);
+            VectorXf context;
+            string choice = this->variation_bandit.choose(root_context);
 
             if (choice == "cx")
             {
                 const Individual<T>& dad = pop[
                     *r.select_randomly(parents.begin(), parents.end())];
 
+
                 // std::cout << "Performing crossover" << std::endl;
                 auto variation_result = cross(mom, dad);
                 ind_parents = {mom, dad};
                 tie(opt, context) = variation_result;
+
+                context = root_context;
             }
             else
             {