From 471ebe7ddc87aaa71ab4205a6bc2d3c55a9bcd6b Mon Sep 17 00:00:00 2001 From: Zahid Irfan Date: Wed, 2 Oct 2024 12:13:17 +0100 Subject: [PATCH] In addition to the fix of training_test condition never true for Multi-objective optimization #166. This was fixed but not pushed in original commit. --- src/utilities/stats/file_io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/utilities/stats/file_io.py b/src/utilities/stats/file_io.py index e7f8bf7b..fbb0a4fd 100644 --- a/src/utilities/stats/file_io.py +++ b/src/utilities/stats/file_io.py @@ -68,7 +68,7 @@ def save_best_ind_to_file(stats, ind, end=False, name="best"): savefile.write("Phenotype:\n" + str(ind.phenotype) + "\n\n") savefile.write("Genotype:\n" + str(ind.genome) + "\n") savefile.write("Tree:\n" + str(ind.tree) + "\n") - if hasattr(params['FITNESS_FUNCTION'], "training_test"): + if params["DATASET_TEST"]: if end: savefile.write("\nTraining fitness:\n" + str(ind.training_fitness)) savefile.write("\nTest fitness:\n" + str(ind.test_fitness))