From f22c97975cced17febeadb453c6e1bf53b708af2 Mon Sep 17 00:00:00 2001 From: Harry Slatyer Date: Sun, 4 Dec 2016 11:09:49 +1100 Subject: [PATCH] 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. --- mloop/learners.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mloop/learners.py b/mloop/learners.py index b759889..8d80d77 100644 --- 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()