Add test for Predictor

[trackerpp.git] / include / PredictorWrapper.h

diff --git a/include/PredictorWrapper.h b/include/PredictorWrapper.h
index 41f818f..c41a8bc 100644 (file)
--- a/include/PredictorWrapper.h
+++ b/include/PredictorWrapper.h
@@ -14,13 +14,13 @@ namespace suanzi {
     class PredictorWrapper
     {
     public:
-        static PredictorWrapperPtr create(const std::string& fname);
+        static PredictorWrapperPtr create(const std::string& python_dir, const std::string& model_dir); // model.pkl file
         ~PredictorWrapper(){}
         void dump();
         double predict(int index, const std::vector<double>& f);
 
     private:
-        PredictorWrapper(const std::string& fname);
+        PredictorWrapper(const std::string& py_dir, const std::string& fname);
         static PredictorWrapperWPtr instance;
 
         PY_FUN dump_func;