fix

models/rank/dnn/config.yaml

@@ -61,3 +61,8 @@ executor:
    dataset_name: dataset_2 # 名字，用来区分不同的阶段
    thread_num: 1 # 线程数
    is_infer: False # 是否是infer
+# - name: infer
+#   model: "{workspace}/model.py" # 模型路径
+#   dataset_name: dataset_2 # 名字，用来区分不同的阶段
+#   thread_num: 1 # 线程数
+#   is_infer: True # 是否是infer

models/rank/dnn/model.py

@@ -77,17 +77,21 @@ class Model(ModelBase):
 
         self.predict = predict
 
-        cost = fluid.layers.cross_entropy(
-            input=self.predict, label=self.label_input)
-        avg_cost = fluid.layers.reduce_mean(cost)
-        self._cost = avg_cost
-
         auc, batch_auc, _ = fluid.layers.auc(input=self.predict,
                                              label=self.label_input,
                                              num_thresholds=2**12,
                                              slide_steps=20)
+        if is_infer:
+            self._infer_results["AUC"] = auc
+            self._infer_results["BATCH_AUC"] = batch_auc
+            return
+
         self._metrics["AUC"] = auc
         self._metrics["BATCH_AUC"] = batch_auc
+        cost = fluid.layers.cross_entropy(
+            input=self.predict, label=self.label_input)
+        avg_cost = fluid.layers.reduce_mean(cost)
+        self._cost = avg_cost
 
     def optimizer(self):
         optimizer = fluid.optimizer.Adam(self.learning_rate, lazy_mode=True)