update eval metric for dssm test cases

alibaba · Feb 2, 2024 · 3ee1556 · 3ee1556
1 parent fbf5b18
commit 3ee1556
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/samples/model_config/dssm_neg_sampler_need_key_feature.config b/samples/model_config/dssm_neg_sampler_need_key_feature.config
@@ -24,8 +24,7 @@ train_config {
 
 eval_config {
   metrics_set: {
-    auc {
-    }
+    recall_at_topk { topk: 3 }
   }
   # currently not supported
   # metrics_set: {

diff --git a/samples/model_config/dssm_neg_sampler_sequence_feature.config b/samples/model_config/dssm_neg_sampler_sequence_feature.config
@@ -24,8 +24,7 @@ train_config {
 
 eval_config {
   metrics_set: {
-    auc {
-    }
+    recall_at_topk { topk: 3 }
   }
   # currently not supported
   # metrics_set: {