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 (new PredictorWrapper("predictor", "../python"));
   9     predictor->load("../resources/model.pkl");
  10     predictor->dump();
  11     std::vector<double> ff (40, 1);
  12     double prob = predictor->predict(4, ff);
  13 }