added way to add deep keys for Patience criterion

climin/stops/stops.py

@@ -247,6 +247,10 @@ class Patience(object):
         i = info['n_iter']
         if isinstance(self.func_or_key, (str, unicode)):
             loss = info[self.func_or_key]
+        elif isinstance(self.func_or_key, (tuple, list)):
+            loss = info
+            for j in self.func_or_key:
+                loss = loss[j]
         else:
             loss = self.func_or_key()
 

test/test_stops.py

@@ -28,3 +28,14 @@ def test_patience_increase():
     assert not stopper({'n_iter': 4})
     assert not stopper({'n_iter': 5})
     assert stopper({'n_iter': 6})
+
+
+def test_patience_deep():
+    vals = iter([{'foo': {'bar': 10}}] * 4).next
+    func = iter([10, 10, 10, 10]).next
+    stopper = Patience(('foo', 'bar'), 3, 2)
+
+    assert not stopper({'n_iter': 0, 'foo': {'bar': 10}})
+    assert not stopper({'n_iter': 1, 'foo': {'bar': 10}})
+    assert not stopper({'n_iter': 2, 'foo': {'bar': 10}})
+    assert stopper({'n_iter': 3, 'foo': {'bar': 10}}), 'initial patience should be over'