changed cross validation wrapper (aimacode#346)

Allen-Hao-Huang · norvig · commit 4fd06ed61048 · 2017-05-28T11:54:31.000-07:00
is supposed to return an answer when errT converges, not errV
used to return size of when err_val converges but is supposed to return the size with minimum err_val
diff --git a/learning.py b/learning.py
@@ -949,13 +949,20 @@ def cross_validation_wrapper(learner, dataset, k=10, trials=1):
     err_val = []
     err_train = []
     size = 1
+    
     while True:
         errT, errV = cross_validation(learner, size, dataset, k)
         # Check for convergence provided err_val is not empty
-        if (err_val and isclose(err_val[-1], errV, rel_tol=1e-6)):
-            best_size = size
-            return learner(dataset, best_size)
-
+        if (err_train and isclose(err_train[-1], errT, rel_tol=1e-6)):
+            best_size = 0
+            min_val = math.inf
+
+            i = 0
+            while i<size:
+                if err_val[i] < min_val:
+                    min_val = err_val[i]
+                    best_size = i
+                i += 1
         err_val.append(errV)
         err_train.append(errT)
         print(err_val)
