(a, b, data_is_normalized=False)
| 11 | |
| 12 | |
| 13 | def _cosine_distance(a, b, data_is_normalized=False): |
| 14 | if not data_is_normalized: |
| 15 | a = np.asarray(a) / np.linalg.norm(a, axis=1, keepdims=True) |
| 16 | b = np.asarray(b) / np.linalg.norm(b, axis=1, keepdims=True) |
| 17 | return 1. - np.dot(a, b.T) |
| 18 | |
| 19 | |
| 20 | def _nn_cosine_distance(x, y): |
no outgoing calls
no test coverage detected