From 366e2ff546092d9be26411fb698b3ddd8e834a11 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期一, 09 十二月 2024 15:38:02 +0800
Subject: [PATCH] 更新实名比对
---
compare/realName.go | 15 +++++++++++++--
1 files changed, 13 insertions(+), 2 deletions(-)
diff --git a/compare/realName.go b/compare/realName.go
index 4753655..d086abd 100644
--- a/compare/realName.go
+++ b/compare/realName.go
@@ -3,14 +3,25 @@
import (
"basic.com/pubsub/protomsg.git"
"sdkCompare/cache"
+ "sort"
)
func realNamePersonsCompere(args protomsg.CompareArgs, fFeature []float32, baseScore float32) protomsg.SdkCompareResult {
var scr protomsg.SdkCompareResult
targets := cache.RealNameDbMap.Walk(DoSdkCompare, fFeature, baseScore)
if len(targets) > 0 {
- for idx, _ := range targets {
- scr.CompareResult = append(scr.CompareResult, targets[idx])
+ sort.Slice(targets, func(i, j int) bool {
+ return targets[i].CompareScore > targets[j].CompareScore
+ })
+
+ // 鎴彇鍓�5涓厓绱�
+ topTargets := targets
+ if len(targets) > 5 {
+ topTargets = targets[:5] // 鍙彇鍓�5涓厓绱�
+ }
+
+ for idx, _ := range topTargets {
+ scr.CompareResult = append(scr.CompareResult, topTargets[idx])
}
}
--
Gitblit v1.8.0