Don't do one last train in order to predict minima

at the end. This was causing an exception to be thrown when trying to get costs from the queue.
michaelhush · f22c97975cced17febeadb453c6e1bf53b708af2 · charmasaur committed Dec 4, 2016 · Dec 4, 2016 · f22c979
1 parent 3e4b3df
commit f22c97975cced17febeadb453c6e1bf53b708af2
Unified Split

Showing with 5 additions and 3 deletions.

+5 −3 mloop/learners.py
diff --git a/mloop/learners.py b/mloop/learners.py
@@ -1904,9 +1904,11 @@ def run(self):
                         raise LearnerInterrupt()
         except LearnerInterrupt:
             pass
-        if self.predict_global_minima_at_end or self.predict_local_minima_at_end:
-            self.get_params_and_costs()
-            self.fit_neural_net()
+        # TODO: Fix this. We can't just do what's here because the costs queue might be empty, but
+        # we should get anything left in it and do one last train.
+        #if self.predict_global_minima_at_end or self.predict_local_minima_at_end:
+        #    self.get_params_and_costs()
+        #    self.fit_neural_net()
         end_dict = {}
         if self.predict_global_minima_at_end:
             self.find_global_minima()