X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;f=test%2FTestPredictor.cpp;h=0b2a6b605252386a0b6b11b0460d8a3a7534abed;hb=97b147e9106cb549dd7dd62829b9d29b326738c7;hp=e3add175f26043789a80f4681e5fc4d2a1d17b37;hpb=8b450dd21387f7775cfc68dee771bda82c0f8ed6;p=trackerpp.git

diff --git a/test/TestPredictor.cpp b/test/TestPredictor.cpp
index e3add17..0b2a6b6 100644
--- a/test/TestPredictor.cpp
+++ b/test/TestPredictor.cpp
@@ -1,7 +1,9 @@
 #include "gtest/gtest.h"
 #include "PredictorWrapper.h"
+#include <iostream>
 
 using namespace suanzi;
+using namespace std;
 
 TEST(Predictor, load)
 {
@@ -10,4 +12,21 @@ TEST(Predictor, load)
     predictor->dump();
     std::vector<double> ff (40, 1);
     double prob = predictor->predict(4, ff);
+    cout << prob << endl;
 }
+
+
+//
+//TEST(Predictor, predict)
+//{
+//    PredictorWrapperPtr predictor (new PredictorWrapper("predictor", "../python"));
+//    predictor->load("../resources/model.pkl");
+//    predictor->dump();
+//    //std::vector<double> ff (40, 1);
+//
+//    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 };
+//    int index = 4;
+//    double prob = predictor->predict(index, ff);
+//    cout << prob << endl;
+//}
+//