丁云鹏 5 meses atrás
pai
commit
7c4cc70da7

+ 1 - 1
recommend-model-produce/src/main/python/models/dnn/config.yaml

@@ -20,7 +20,7 @@ runner:
   train_data_dir: "data/sample_data/train"
   train_reader_path: "criteo_reader" # importlib format
   use_gpu: False
-  use_auc: True
+  use_auc: False
   train_batch_size: 2
   epochs: 1
   print_interval: 2

+ 3 - 1
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DNNModel.java

@@ -38,7 +38,9 @@ public class DNNModel implements Model {
         float[] outData = new float[outHandle.getSize()];
         outHandle.copyToCpu(outData);
         long time2 = System.currentTimeMillis();
-        log.info("predictor2 outData[0]={},outDataLen={},cost={}", outData[0], outData.length, (time2 - time1));
+        log.info("predictor2 inNames={},outNames={},outData[0]={},outDataLen={},cost={}", inNames, outNames,outData[0],
+                outData.length,
+                (time2 - time1));
 
         outHandle.destroyNativeTensor();
         inHandle.destroyNativeTensor();