Merge pull request #113 from perib/dev

fixed recursive search space probabilities
EpistasisLab · Dec 5, 2023 · 1fc2a76 · 1fc2a76
2 parents cf5d1af + a904acf
commit 1fc2a76
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tpot2/individual_representations/graph_pipeline_individual/individual.py b/tpot2/individual_representations/graph_pipeline_individual/individual.py
@@ -1193,7 +1193,10 @@ def random_weighted_sort(l,weights, rng_=None):
     sorted_l = []
     indeces = {i: weights[i] for i in range(len(l))}
     while len(indeces) > 0:
-        next_item = rng.choice(list(indeces.keys()), p=list(indeces.values()))
+        keys = list(indeces.keys())
+        p = np.array([indeces[k] for k in keys])
+        p = p / p.sum()
+        next_item = rng.choice(list(indeces.keys()), p=p)
         indeces.pop(next_item)
         sorted_l.append(l[next_item])