Browse Source

Rename distance to score (#948)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 1 year ago
parent
commit
bd6c41d75b

+ 1 - 0
.gitignore

@@ -31,6 +31,7 @@ volumes/
 *.iml
 
 # Example files
+examples/bulk_writer
 examples/main/java/io/milvus/tls/*
 !examples/main/java/io/milvus/tls/gen.sh
 !examples/main/java/io/milvus/tls/openssl.cnf

+ 1 - 1
src/main/java/io/milvus/orm/iterator/SearchIterator.java

@@ -421,7 +421,7 @@ public class SearchIterator {
     }
 
     private float getDistance(QueryResultsWrapper.RowRecord record) {
-        return (float) record.get("distance");
+        return (float) record.get("score");
     }
 
     // Avoiding precision loss when converting to exponent notation.

+ 2 - 2
src/main/java/io/milvus/response/SearchResultsWrapper.java

@@ -87,7 +87,7 @@ public class SearchResultsWrapper extends RowRecordWrapper {
                 record.put("id", score.getStrID());
             }
 
-            record.put("distance", score.getScore());
+            record.put("score", score.getScore());
             buildRowRecord(record, i);
             records.add(record);
         }
@@ -156,7 +156,7 @@ public class SearchResultsWrapper extends RowRecordWrapper {
 
         List<IDScore> idScores = new ArrayList<>();
 
-        // set id and distance
+        // set id and score
         IDs ids = results.getIds();
         if (ids.hasIntId()) {
             LongArray longIDs = ids.getIntId();

+ 1 - 1
src/main/java/io/milvus/v2/service/vector/response/SearchResp.java

@@ -34,7 +34,7 @@ public class SearchResp {
     @SuperBuilder
     public static class SearchResult {
         private Map<String, Object> entity;
-        private Float distance;
+        private Float score;
         private Object id;
     }
 }

+ 1 - 1
src/main/java/io/milvus/v2/utils/ConvertUtils.java

@@ -65,7 +65,7 @@ public class ConvertUtils {
         for (int i = 0; i < numQueries; i++) {
             searchResults.add(searchResultsWrapper.getIDScore(i).stream().map(idScore -> SearchResp.SearchResult.builder()
                     .entity(idScore.getFieldValues())
-                    .distance(idScore.getScore())
+                    .score(idScore.getScore())
                     .id(idScore.getStrID().isEmpty() ? idScore.getLongID() : idScore.getStrID())
                     .build()).collect(Collectors.toList()));
         }