projects
/
trackerpp.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Add test for Predictor
[trackerpp.git]
/
include
/
PredictorWrapper.h
diff --git
a/include/PredictorWrapper.h
b/include/PredictorWrapper.h
index
9ceb21a
..
c41a8bc
100644
(file)
--- a/
include/PredictorWrapper.h
+++ b/
include/PredictorWrapper.h
@@
-14,13
+14,13
@@
namespace suanzi {
class PredictorWrapper
{
public:
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();
~PredictorWrapper(){}
void dump();
- double predict(int index,
std::vector<double>
f);
+ double predict(int index,
const std::vector<double>&
f);
private:
private:
- PredictorWrapper(const std::string& fname);
+ PredictorWrapper(const std::string&
py_dir, const std::string&
fname);
static PredictorWrapperWPtr instance;
PY_FUN dump_func;
static PredictorWrapperWPtr instance;
PY_FUN dump_func;