From 5968a18dab59806c5b8c131d40c7fc965a227b45 Mon Sep 17 00:00:00 2001 From: perib Date: Mon, 7 Aug 2023 10:08:49 -0700 Subject: [PATCH] fix get_pareto_front --- tpot2/utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tpot2/utils/utils.py b/tpot2/utils/utils.py index 7963eb03..fe165993 100644 --- 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=[])