X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;f=python%2Fpredictor.py;h=2b77aac135d53e9004a8a2739d3b5794bfdee5bb;hb=209cfd9fe0ea398b794d1d1995629a826125f035;hp=9f21518b92a4e79a3d090f9607954abfe4058830;hpb=48adce31a0ffdb3757ee1be8a63ce7e769e87deb;p=trackerpp.git

diff --git a/python/predictor.py b/python/predictor.py
index 9f21518..2b77aac 100644
--- a/python/predictor.py
+++ b/python/predictor.py
@@ -15,15 +15,18 @@ def init(fname = './model.pkl'):
 
 def dump():
     global predictors
+    ss = '\n'
     for i in predictors:
-        print i
-        print i.coef_
+        ss += str(i.__dict__)
+        ss += '\n'
+    return ss
 
 
 def predict(index, features):
     pp = predictors[index]
     true_class = int(pp.classes_[1] == 1)
     prob = pp.predict_proba([features])[0, true_class]
+    print prob
     return prob