Update compare function in learning.py

lucasmoura · lucasmoura · commit 1af3c5249964 · 2017-04-07T11:12:13.000-03:00
diff --git a/learning.py b/learning.py
@@ -1000,14 +1000,14 @@ def ContinuousXor(n):
 # ______________________________________________________________________________
 
 
-def compare(algorithms=[PluralityLearner, NaiveBayesLearner,
-                        NearestNeighborLearner, DecisionTreeLearner],
-            datasets=[iris, orings, zoo, restaurant, SyntheticRestaurant(20),
-                      Majority(7, 100), Parity(7, 100), Xor(100)],
-            k=10, trials=1):
+def compare(algorithms=[PluralityLearner, NaiveBayesLearner, NearestNeighborLearner,
+                        DecisionTreeLearner],
+            datasets=[iris, orings, zoo, restaurant, SyntheticRestaurant(20), Majority(7, 100),
+                      Parity(7, 100), Xor(100)],
+            k=10, size=3, trials=1):
     """Compare various learners on various datasets using cross-validation.
     Print results as a table."""
     print_table([[a.__name__.replace('Learner', '')] +
-                 [cross_validation(a, d, k, trials) for d in datasets]
+                 [cross_validation(a, size, d, k, trials) for d in datasets]
                  for a in algorithms],
-                header=[''] + [d.name[0:7] for d in datasets], numfmt='%.2f')
+                 header=[''] + [d.name[0:7] for d in datasets], numfmt='{:.2f}')
