X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;f=python%2Fpredictor.py;h=2b77aac135d53e9004a8a2739d3b5794bfdee5bb;hb=209cfd9fe0ea398b794d1d1995629a826125f035;hp=b58fd41a360311ac0b47753003f722fc4ca18d87;hpb=3ff9a5ad691b8dca9d91f8e9786a8d08d31b70fa;p=trackerpp.git diff --git a/python/predictor.py b/python/predictor.py index b58fd41..2b77aac 100644 --- 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