fix get_pareto_front

EpistasisLab · Aug 7, 2023 · 5968a18 · 5968a18
1 parent 58ac7f6
commit 5968a18
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=[])