Bug fix for stability exit when no validation set is used.

aigamedev · Nov 21, 2015 · e1c4f4b · e1c4f4b
1 parent 66ba3fb
commit e1c4f4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sknn/mlp.py b/sknn/mlp.py
@@ -184,7 +184,7 @@ def _train(self, X, y):
                 log.info("User defined callback terminated at %i iterations.", i)
                 break
 
-            if self.valid_set is not None and n_stable >= self.n_stable:
+            if self.n_stable is not None and n_stable >= self.n_stable:
                 log.debug("")
                 log.info("Early termination condition fired at %i iterations.", i)
                 break