#include "SharedPtr.h"
#include <boost/python.hpp>
+#include <vector>
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() { this->dump_func(); }
- void predict() { this->predict_func();}
+ 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;