class PredictorWrapper
{
public:
- static PredictorWrapperPtr create(const std::string& python_dir, const std::string& model_dir); // model.pkl file
+ PredictorWrapper(const std::string& module="predictor", const std::string& pydir = "./python");
~PredictorWrapper(){}
void dump();
double predict(int index, const std::vector<double>& f);
+ bool load(const std::string& fname); // load pkl file
private:
- PredictorWrapper(const std::string& py_dir, const std::string& fname);
- static PredictorWrapperWPtr instance;
-
- PY_FUN dump_func;
- PY_FUN predict_func;
-
+ boost::python::object m_module;
};
}