init

ad-engine-commons/src/main/java/com/tzld/piaoquan/ad/engine/commons/score/model/XGBoostModel.java

@@ -2,7 +2,6 @@ package com.tzld.piaoquan.ad.engine.commons.score.model;
 
 
 import com.tzld.piaoquan.ad.engine.commons.util.CompressUtil;
-import ml.dmlc.xgboost4j.java.DMatrix;
 import ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel;
 import org.apache.commons.lang.math.NumberUtils;
 import org.apache.spark.ml.linalg.Vector;
@@ -77,11 +76,10 @@ public class XGBoostModel extends Model {
 
     @Override
     public boolean loadFromStream(InputStream in) throws Exception {
-        String modelDir = "";
+        String modelDir = "model2";
         CompressUtil.decompressGzFile(in, modelDir);
         XGBoostClassificationModel model2 = XGBoostClassificationModel.load("file://" + modelDir);
-        model2.setMissing(0.0f)
-                .setFeaturesCol("features");
+        model2.setMissing(0.0f);
         this.model = model2;
         return true;
     }