Update query_rewriter.go

cosine and dot into distance
dgraph-io · May 7, 2024 · f1ece2b · f1ece2b
1 parent 117de1b
commit f1ece2b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/graphql/resolve/query_rewriter.go b/graphql/resolve/query_rewriter.go
@@ -662,9 +662,9 @@ func rewriteAsSimilarByIdQuery(
 	distanceFormula := "math((v2 - v1) dot (v2 - v1))" // default - euclidian
 
 	if metric == schema.SimilarSearchMetricDotProduct {
-		distanceFormula = "math(v1 dot v2)"
+		distanceFormula = "math((1.0 - (v1 dot v2)) /2.0)"
 	} else if metric == schema.SimilarSearchMetricCosine {
-		distanceFormula = "math((v1 dot v2) / ((v1 dot v1) * (v2 dot v2)))"
+		distanceFormula = "math((1.0 - ((v1 dot v2) / ((v1 dot v1) * (v2 dot v2)))) / 2.0)"
 	}
 
 	// First generate the query to fetch the uid
@@ -822,7 +822,7 @@ func rewriteAsSimilarByEmbeddingQuery(
 	distanceFormula := "math((v2 - $search_vector) dot (v2 - $search_vector))" // default = euclidian
 
 	if metric == schema.SimilarSearchMetricDotProduct {
-		distanceFormula = "math(($search_vector) dot v2)"
+		distanceFormula = "math(( 1.0 - (($search_vector) dot v2)) /2.0)"
 	} else if metric == schema.SimilarSearchMetricCosine {
 		distanceFormula = "math(( ($search_vector) dot v2) / (( ($search_vector) dot ($search_vector)) * (v2 dot v2)))"
 	}