test/TestPredictor.cpp

   1 #include "gtest/gtest.h"
   2 #include "PredictorWrapper.h"
   3 #include <iostream>
   4
   5 using namespace suanzi;
   6 using namespace std;
   7
   8 TEST(Predictor, load)
   9 {
  10     PredictorWrapperPtr predictor (new PredictorWrapper("predictor", "../python"));
  11     predictor->load("../resources/model.pkl");
  12     predictor->dump();
  13     std::vector<double> ff (40, 1);
  14     double prob = predictor->predict(4, ff);
  15     cout << prob << endl;
  16 }
  17
  18
  19 //
  20 //TEST(Predictor, predict)
  21 //{
  22 //    PredictorWrapperPtr predictor (new PredictorWrapper("predictor", "../python"));
  23 //    predictor->load("../resources/model.pkl");
  24 //    predictor->dump();
  25 //    //std::vector<double> ff (40, 1);
  26 //
  27 //    std::vector<double> ff = { 6.9154247185294e-310, 6.91542471861144e-310, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 6.9154247185294e-310, 6.91542471861144e-310, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, -3.0949108904043924e+217, -3.0949108904066356e+217, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 6.91542471837167e-310, 6.9154247185979e-310, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 6.9154247183334e-310, 6.9154247185946e-310, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0 };
  28 //    int index = 4;
  29 //    double prob = predictor->predict(index, ff);
  30 //    cout << prob << endl;
  31 //}
  32 //