return zero recall on empty ground truth (#381)

gorse-io · Feb 14, 2022 · f9df387 · f9df387
1 parent df2feef
commit f9df387
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/base/search/hnsw.go b/base/search/hnsw.go
@@ -349,6 +349,9 @@ func (b *HNSWBuilder) evaluate(idx *HNSW, prune0 bool) float32 {
 		}
 		return nil
 	})
+	if count == 0 {
+		return 0
+	}
 	return result / count
 }
 

diff --git a/base/search/ivf.go b/base/search/ivf.go
@@ -302,6 +302,9 @@ func (b *IVFBuilder) evaluate(idx *IVF, prune0 bool) float32 {
 		}
 		return nil
 	})
+	if count == 0 {
+		return 0
+	}
 	return result / count
 }