--- a/python/ml.py	Fri May 23 17:33:11 2014 -0400
+++ b/python/ml.py	Wed May 28 17:46:38 2014 -0400
@@ -24,8 +24,8 @@
     def train(self, samples, responses):
         self.model.train(samples, responses, params = self.params)
 
-    def predict(self, sample):
-        return np.float32(self.model.predict(s))
+    def predict(self, samples):
+        return np.float32([self.model.predict(s) for s in samples])
 
 
 class Centroid: