github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-04 04:28:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Suppress divide by zero wranings for cosine similarity (#9006)

Browse Source 
Suppress run time warnings for divide by zero as the downstream code
handles the scenario (handling inf and nan)
This commit is contained in:
Eugene Yurtsev
2023-08-09 18:56:51 -04:00
committed by GitHub
parent 5454591b0a
commit efa02ed768
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/utils/math.py
View File

@@ -20,7 +20,9 @@ def cosine_similarity(X: Matrix, Y: Matrix) -> np.ndarray:
X_norm = np.linalg.norm(X, axis=1) X_norm = np.linalg.norm(X, axis=1)
Y_norm = np.linalg.norm(Y, axis=1) Y_norm = np.linalg.norm(Y, axis=1)
similarity = np.dot(X, Y.T) / np.outer(X_norm, Y_norm) # Ignore divide by zero errors run time warnings as those are handled below.
with np.errstate(divide="ignore", invalid="ignore"):
similarity = np.dot(X, Y.T) / np.outer(X_norm, Y_norm)
similarity[np.isnan(similarity) | np.isinf(similarity)] = 0.0 similarity[np.isnan(similarity) | np.isinf(similarity)] = 0.0
return similarity return similarity