projects
/
trackerpp.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Fix nan issue in features
[trackerpp.git]
/
include
/
PredictorWrapper.h
diff --git
a/include/PredictorWrapper.h
b/include/PredictorWrapper.h
index
c41a8bc
..
18ea862
100644
(file)
--- a/
include/PredictorWrapper.h
+++ b/
include/PredictorWrapper.h
@@
-14,18
+14,14
@@
namespace suanzi {
class PredictorWrapper
{
public:
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);
~PredictorWrapper(){}
void dump();
double predict(int index, const std::vector<double>& f);
+ bool load(const std::string& fname); // load pkl file
private:
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;
};
}
};
}