Merge pull request #340 from rhiever/master

Fixed #339
EpistasisLab · Jan 5, 2017 · a24454c · a24454c
2 parents 8836be0 + 7440558
commit a24454c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tpot/base.py b/tpot/base.py
@@ -385,6 +385,8 @@ def pareto_eq(ind1, ind2):
                 for pipeline, pipeline_scores in zip(self._pareto_front.items, reversed(self._pareto_front.keys)):
                     if pipeline_scores.wvalues[1] > top_score:
                         self._optimized_pipeline = pipeline
+                        top_score = pipeline_scores.wvalues[1]
+
                 # It won't raise error for a small test like in a unit test becasue a few pipeline sometimes
                 # may fail due to the training data does not fit the operator's requirement. 
                 if self.generations*self.population_size > 5 and not self._optimized_pipeline: