upgrade opencv to 3.3.1, complete patch similarity

[trackerpp.git] / python / predictor.py

diff --git a/python/predictor.py b/python/predictor.py
index 9f21518..2b77aac 100644 (file)
--- 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