separate pytwrpper from predictorWrappery

[trackerpp.git] / test / TestPredictor.cpp

diff --git a/test/TestPredictor.cpp b/test/TestPredictor.cpp
index c6087f0..e3add17 100644 (file)
--- a/test/TestPredictor.cpp
+++ b/test/TestPredictor.cpp
@@ -5,7 +5,8 @@ using namespace suanzi;
 
 TEST(Predictor, load)
 {
-    PredictorWrapperPtr predictor = PredictorWrapper::create("../python", "../python/model.pkl");
+    PredictorWrapperPtr predictor (new PredictorWrapper("predictor", "../python"));
+    predictor->load("../resources/model.pkl");
     predictor->dump();
     std::vector<double> ff (40, 1);
     double prob = predictor->predict(4, ff);