Merge pull request #7 from mswimmer/master

Make tagged VW data work
josephreisinger · Sep 16, 2013 · c0da0cc · c0da0cc
2 parents 591d95e + 7224d40
commit c0da0cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vowpal_porpoise/vw.py b/vowpal_porpoise/vw.py
@@ -233,7 +233,7 @@ def parse_prediction(self, p):
         if self.lda:
             return map(float, p.split())
         else:
-            return float(p)
+            return float(p.split()[0])
 
     def read_predictions_(self):
         for x in open(self.prediction_file):