projects
/
trackerpp.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
| inline |
side by side
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
b58fd41
..
2b77aac
100644
(file)
--- a/
python/predictor.py
+++ b/
python/predictor.py
@@
-19,8
+19,6
@@
def dump():
for i in predictors:
ss += str(i.__dict__)
ss += '\n'
-# ss += str(i.coef_)
-# ss += '\n'
return ss
@@
-28,6
+26,7
@@
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