topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 20:51:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvment)(chat) remove the parse in candidate parse that has the same score as selectparse (#201)

Browse Source 
Co-authored-by: jolunoluo
This commit is contained in:
LXW
2023-10-13 10:05:51 +08:00
committed by GitHub
parent 26beff1080
commit 7544780ff7
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java
View File

@@ -160,7 +160,12 @@ public class QueryServiceImpl implements QueryService {
return candidateParses; return candidateParses;
} }
int selectParseSize = selectedParses.size(); int selectParseSize = selectedParses.size();
Set<Double> selectParseScoreSet = selectedParses.stream()
.map(SemanticParseInfo::getScore).collect(Collectors.toSet());
int candidateParseSize = 5 - selectParseSize; int candidateParseSize = 5 - selectParseSize;
candidateParses = candidateParses.stream()
.filter(candidateParse -> !selectParseScoreSet.contains(candidateParse.getScore()))
.collect(Collectors.toList());
SemanticParseInfo semanticParseInfo = selectedParses.get(0); SemanticParseInfo semanticParseInfo = selectedParses.get(0);
Long modelId = semanticParseInfo.getModelId(); Long modelId = semanticParseInfo.getModelId();
if (modelId == null || modelId <= 0) { if (modelId == null || modelId <= 0) {