test/TestPredictor.cpp

   1 #include "gtest/gtest.h"
   2 #include "PredictorWrapper.h"
   3
   4 using namespace suanzi;
   5
   6 TEST(Predictor, load)
   7 {
   8     PredictorWrapperPtr predictor = PredictorWrapper::create("../python", "../python/model.pkl");
   9     predictor->dump();
  10     std::vector<double> ff (40, 1);
  11     double prob = predictor->predict(4, ff);
  12 }