We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7cb3d55 commit 0a70173Copy full SHA for 0a70173
bigframes/bigquery/_operations/search.py
@@ -164,12 +164,12 @@ def vector_search(
164
... query=search_query,
165
... distance_type="cosine",
166
... query_column_to_search="another_embedding",
167
- ... top_k=2)
+ ... top_k=2).sort_values("id")
168
query_id embedding another_embedding id my_embedding distance
169
- 1 cat [3. 5.2] [3.3 5.2] 2 [2. 4.] 0.005181
170
- 0 dog [1. 2.] [0.7 2.2] 4 [1. 3.2] 0.000013
171
1 cat [3. 5.2] [3.3 5.2] 1 [1. 2.] 0.005181
+ 1 cat [3. 5.2] [3.3 5.2] 2 [2. 4.] 0.005181
172
0 dog [1. 2.] [0.7 2.2] 3 [1.5 7. ] 0.004697
+ 0 dog [1. 2.] [0.7 2.2] 4 [1. 3.2] 0.000013
173
<BLANKLINE>
174
[4 rows x 6 columns]
175
0 commit comments