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: {