Update StochasticHillClimb.py

100 · Jul 6, 2017 · 9cb1c8a · 9cb1c8a
1 parent 3c2ccc3
commit 9cb1c8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Solid/StochasticHillClimb.py b/Solid/StochasticHillClimb.py
@@ -121,11 +121,11 @@ def run(self, verbose=True):
             if self._accept_neighbor(neighbor):
                 self.current_state = neighbor
 
-            if self.best_objective is None or self._objective(self.current_state) > self.best_objective:
+            if self._objective(self.current_state) > (self.best_objective or 0):
                 self.best_objective = self._objective(self.current_state)
                 self.best_state = deepcopy(self.current_state)
 
-            if self.best_objective is not None and self.max_objective is not None and self.best_objective > self.max_objective:
+            if self.max_objective is not None and (self.best_objective or 0) > self.max_objective:
                 print("TERMINATING - REACHED MAXIMUM OBJECTIVE")
                 return self.best_state, self.best_objective
         print("TERMINATING - REACHED MAXIMUM STEPS")