From 3ee1556d8603b13f72957fa14278f2f3fa614a2f Mon Sep 17 00:00:00 2001
From: chengmengli06 <chengmengli06@qq.com>
Date: Fri, 2 Feb 2024 14:47:22 +0800
Subject: [PATCH] update eval metric for dssm test cases

---
 samples/model_config/dssm_neg_sampler_need_key_feature.config | 3 +--
 samples/model_config/dssm_neg_sampler_sequence_feature.config | 3 +--
 2 files changed, 2 insertions(+), 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
index f1b52031d..ea7cc8342 100644
--- 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
index 6038b8952..5ce0af37f 100644
--- 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: {