Merge pull request #35 from perib/dev

fix get_pareto_front
EpistasisLab · Aug 7, 2023 · 8d2c878 · 8d2c878
2 parents 58ac7f6 + 5968a18
commit 8d2c878
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tpot2/utils/utils.py b/tpot2/utils/utils.py
@@ -112,7 +112,7 @@ def get_pareto_front(df, column_names, weights, invalid_values=["TIMEOUT","INVAL
     indexes = dftmp[~dftmp[column_names].isna().any(axis=1)].index.values
     weighted_scores = df.loc[indexes][column_names].to_numpy()  * weights
 
-    pareto_fronts = tpot2.parent_selectors.nondominated_sorting(weighted_scores)
+    pareto_fronts = tpot2.selectors.nondominated_sorting(weighted_scores)
 
     df = pd.DataFrame(index=df.index,columns=["Pareto_Front"], data=[])