X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;f=include%2FPredictorWrapper.h;h=9ceb21ac29e4f1c95f17b7867a4f4064c9dc7feb;hb=3aa517d206c44156fe86697aeadc5f75ea212329;hp=ac54f733be99936cf96c1e0bb8e5ba93bb6e97c6;hpb=3ff9a5ad691b8dca9d91f8e9786a8d08d31b70fa;p=trackerpp.git

diff --git a/include/PredictorWrapper.h b/include/PredictorWrapper.h
index ac54f73..9ceb21a 100644
--- a/include/PredictorWrapper.h
+++ b/include/PredictorWrapper.h
@@ -3,6 +3,7 @@
 
 #include "SharedPtr.h"
 #include <boost/python.hpp>
+#include <vector>
 
 namespace suanzi {
 
@@ -16,7 +17,7 @@ namespace suanzi {
         static PredictorWrapperPtr create(const std::string& fname);
         ~PredictorWrapper(){}
         void dump();
-        double predict();
+        double predict(int index, std::vector<double> f);
 
     private:
         PredictorWrapper(const std::string& fname);